diff --git a/assignment1/test.nix b/assignment1/test.nix
new file mode 100644
index 0000000..0967ef4
--- /dev/null
+++ b/assignment1/test.nix
@@ -0,0 +1 @@
+{}
diff --git a/flake.nix b/flake.nix
index d13942d..31faf94 100644
--- a/flake.nix
+++ b/flake.nix
@@ -13,15 +13,18 @@
       ];
 
       eachSystem = nixpkgs.lib.genAttrs (import systems);
-    in {
-      packages = eachSystem (system:
+
+      importFromSubdirs = file: eachSystem (system:
         let pkgs = nixpkgs.legacyPackages.${system};
         in pkgs.lib.mergeAttrsList
           (builtins.map (d:
             builtins.mapAttrs
               (_: v: pkgs.callPackage v {})
-              (import ./${d}))
+              (import ./${d}/${file}))
             subdirs));
+    in {
+      packages = importFromSubdirs "default.nix";
+      checks = importFromSubdirs "test.nix";
 
       devShells = eachSystem (system:
         let pkgs = nixpkgs.legacyPackages.${system};