diff --git a/checks/default.nix b/checks/default.nix
index 66d5b0b..81883ba 100644
--- a/checks/default.nix
+++ b/checks/default.nix
@@ -4,7 +4,7 @@
   pkgs,
   deployPkgs,
   ...
-}:
+}@inputs:
 {
   ${system} = {
 
diff --git a/checks/nixos-modules/default.nix b/checks/nixos-modules/default.nix
index 74e8a25..ea8eace 100644
--- a/checks/nixos-modules/default.nix
+++ b/checks/nixos-modules/default.nix
@@ -54,7 +54,7 @@ let
 in
 pipe modulesBaseDir [
   getSubDirs
-  (filter (isFolderWithFile "test.nix" modulesBaseDir))
+  (filter (isFolderWithFile "test.nix"))
   (map mkTest)
   (linkFarmFromDrvs "nixos-modules")
 ]
diff --git a/lib/default.nix b/lib/default.nix
index e4cd3ee..8df2940 100644
--- a/lib/default.nix
+++ b/lib/default.nix
@@ -7,20 +7,16 @@ let
     pathExists
     path
     ;
-  # Get a list of all subdirectories of a directory.
-  getSubDirs = base: attrNames (filterAttrs (n: t: t == "directory") (builtins.readDir base));
-  # Check if a folder with a base path and folder name contains a file with a specific name
   isFolderWithFile =
     fileName: basePath: folderName:
-    (pathExists (path.append basePath "./${folderName}/${fileName}"));
-  # Get a list of subfolders that contain a default.nix file.
-  foldersWithNix = base: filter (isFolderWithFile "default.nix" base) (getSubDirs base);
+    (pathExists (path.append basePath "./${folderName}/${fileName}.nix"));
+  getSubDirs = base: attrNames (filterAttrs (n: t: t == "directory") (builtins.readDir base));
+  foldersWithNix = base: filter (isFolderWithFile "default.nix") (getSubDirs base);
 
 in
 {
-  inherit getSubDirs isFolderWithFile foldersWithNix;
+  inherit isFolderWithFile foldersWithNix;
 
-  # Get a list of default.nix files that are nix submodules of the current folder.
   loadSubmodulesFrom =
     basePath: map (folder: path.append basePath "./${folder}/default.nix") (foldersWithNix basePath);
 }
diff --git a/nixos-modules/static-page/test.py b/nixos-modules/static-page/test.py
index 5c273d5..0dc69fc 100644
--- a/nixos-modules/static-page/test.py
+++ b/nixos-modules/static-page/test.py
@@ -37,7 +37,7 @@ def test(subtest, webserver):
         expect_http_content(webserver, indexContent,
                             "http://localhost/index.html")
 
-    with subtest("example.com is a hosts alias and redirect"):
+    with subtest("example.com is a hosts alias and redirects to localhost"):
         webserver.succeed("grep example.com /etc/hosts")
 
         url = "http://example.com/index.html"