diff --git a/test.nix b/test.nix
index 235f9fc..965b63c 100644
--- a/test.nix
+++ b/test.nix
@@ -1,7 +1,11 @@
 { pkgs ? import <nixpkgs> { overlays = [ (import ./default.nix) ]; } }:
-
+let
+  mypython = pkgs.python3.withPackages (python-pkgs: [
+    python-pkgs.salt-lint
+  ]);
+in 
 pkgs.mkShell {
   buildInputs = with pkgs; [
-    gohome
+    mypython
   ];
 }