diff --git a/diskimage_builder/elements/dracut-regenerate/package-installs.yaml b/diskimage_builder/elements/dracut-regenerate/package-installs.yaml
new file mode 100644
index 000000000..9ef4c76c0
--- /dev/null
+++ b/diskimage_builder/elements/dracut-regenerate/package-installs.yaml
@@ -0,0 +1,8 @@
+PyYAML:
+  phase: pre-install.d
+  dib_python_version: 2
+python3-PyYAML:
+  phase: pre-install.d
+  installtype: package
+  dib_python_version: 3
+
diff --git a/diskimage_builder/elements/dracut-regenerate/pkg-map b/diskimage_builder/elements/dracut-regenerate/pkg-map
new file mode 100644
index 000000000..2249c2572
--- /dev/null
+++ b/diskimage_builder/elements/dracut-regenerate/pkg-map
@@ -0,0 +1,23 @@
+{
+  "family": {
+    "gentoo": {
+      "PyYAML": "dev-python/pyyaml"
+    },
+    "suse": {
+      "PyYAML": "python-PyYAML",
+      "python3-PyYAML": "python3-PyYAML"
+    },
+    "redhat": {
+      "PyYAML": "PyYAML",
+      "python3-PyYAML": "python3-PyYAML"
+    },
+    "debian": {
+      "PyYAML": "python-yaml",
+      "python3-PyYAML": "python3-yaml"
+    }
+  },
+  "default": {
+    "PyYAML": "",
+    "python3-PyYAML": ""
+  }
+}