diff --git a/tests/default.nix b/tests/default.nix
index c10cf76..c7c15a5 100644
--- a/tests/default.nix
+++ b/tests/default.nix
@@ -20,7 +20,7 @@ let
         (lib.attrNames (builtins.readDir ./.))
     );
 
-  allTests = lib.genAttrs (allTestFilenames) (test: import (./. + "/${test}.nix") { inherit makeDiskoTest; }) //
+  allTests = lib.genAttrs (allTestFilenames) (test: import (./. + "/${test}.nix") { inherit makeDiskoTest pkgs; }) //
              evalTest "lvm-luks-example" ../example/config.nix // {
                standalone = (pkgs.nixos [ ../example/stand-alone/configuration.nix ]).config.system.build.toplevel;
              };
diff --git a/tests/swap.nix b/tests/swap.nix
index d3268dc..4861f8d 100644
--- a/tests/swap.nix
+++ b/tests/swap.nix
@@ -16,4 +16,7 @@ makeDiskoTest {
         '
     """);
   '';
+  extraConfig = {
+    environment.systemPackages = [ pkgs.jq ];
+  };
 }