diff --git a/flake-modules/tests.nix b/flake-modules/tests.nix
index 5c0a39ca..9ed0ea4b 100644
--- a/flake-modules/tests.nix
+++ b/flake-modules/tests.nix
@@ -1,6 +1,5 @@
 {
   self,
-  lib,
   helpers,
   ...
 }:
@@ -9,18 +8,14 @@
     {
       pkgs,
       pkgsUnfree,
-      system,
       ...
     }:
     {
-      checks = import ../tests {
+      checks = pkgs.callPackages ../tests {
         inherit
           helpers
-          lib
-          pkgs
           pkgsUnfree
           self
-          system
           ;
       };
     };
diff --git a/tests/default.nix b/tests/default.nix
index 93484f18..f24b55d5 100644
--- a/tests/default.nix
+++ b/tests/default.nix
@@ -3,8 +3,8 @@
   pkgsUnfree,
   helpers,
   lib,
-  system,
   self, # The flake instance
+  system ? pkgs.stdenv.hostPlatform.system,
 }:
 let
   autoArgs = pkgs // {