diff --git a/flake.nix b/flake.nix
index 4228ab3..9696fd9 100644
--- a/flake.nix
+++ b/flake.nix
@@ -22,7 +22,9 @@
 				config.allowUnfree = true;
 			};
 			packages = {
-				vm-test = pkgs.dockerTools.buildLayeredImage {
+				vm-test = let
+					basicPython = (pkgs.python3.withPackages (p: with p; [ pyyaml ]));
+				in pkgs.dockerTools.buildLayeredImage {
 					name = "vm-test";
 					tag = "latest";
 					contents = with pkgs; [
@@ -33,9 +35,8 @@
 						findutils
 						jq
 						packer
-						(python3.withPackages (p: with p; [ pyyaml ]))
 						shellcheck
-						xonsh
+						(xonsh.overridePythonAttrs { python3 = basicPython; })
 					];
 				};
 			};