diff --git a/default.nix b/default.nix
index 9bab938..4c6ced4 100644
--- a/default.nix
+++ b/default.nix
@@ -19,7 +19,8 @@ in
       program =
         (pkgs.writeScript "nix-unit" ''
           #!/usr/bin/env bash
-          ${nix-unit}/bin/nix-unit --flake '.#tests'
+          TEST_TARGET="tests''${1:+.}''${1}";
+          ${nix-unit}/bin/nix-unit --flake ".#''${TEST_TARGET}";
         '').outPath;
     };
   };