diff --git a/flake.nix b/flake.nix
index 61d4d85..00650b3 100644
--- a/flake.nix
+++ b/flake.nix
@@ -23,7 +23,7 @@
 			};
 			packages = {
 				vm-test = let
-					basicPython = (pkgs.python3.withPackages (p: with p; [ ruamel-yaml ]));
+					basicPython = (pkgs.python3.withPackages (p: with p; [ pyyaml ]));
 				in pkgs.dockerTools.buildLayeredImage {
 					name = "vm-test";
 					tag = "latest";
@@ -36,7 +36,7 @@
 						jq
 						packer
 						shellcheck
-						(xonsh.overridePythonAttrs (old: rec { python3 = basicPython; propagatedBuildInputs = old.propagatedBuildInputs ++ [ python3.pkgs.ruamel-yaml ];}))
+						(xonsh.overridePythonAttrs (old: rec { python3 = basicPython; propagatedBuildInputs = old.propagatedBuildInputs ++ [ basicPython.pkgs.pyyaml ];}))
 					];
 				};
 			};