diff --git a/dracut/modules.d/90wwinit/parse-wwinit.sh b/dracut/modules.d/90wwinit/parse-wwinit.sh
index 772a1baf..6115c896 100755
--- a/dracut/modules.d/90wwinit/parse-wwinit.sh
+++ b/dracut/modules.d/90wwinit/parse-wwinit.sh
@@ -7,12 +7,12 @@ if [ "${root}" = "wwinit" ]
 then
     info "root=${root}"
     wwinit_uri="$(getarg wwinit.uri)"
-    export wwinit_container="${wwinit_uri}container&compress=gz"; info "wwinit_container=${wwinit_container}"
-    export wwinit_system="${wwinit_uri}system&compress=gz"; info "wwinit_system=${wwinit_system}"
-    export wwinit_runtime="${wwinit_uri}runtime&compress=gz"; info "wwinit_runtime=${wwinit_runtime}"
+    export wwinit_container="${wwinit_uri}&stage=container&compress=gz"; info "wwinit_container=${wwinit_container}"
+    export wwinit_system="${wwinit_uri}&stage=system&compress=gz"; info "wwinit_system=${wwinit_system}"
+    export wwinit_runtime="${wwinit_uri}&stage=runtime&compress=gz"; info "wwinit_runtime=${wwinit_runtime}"
     if [ -n "$(getarg wwinit.KernelOverride)" ]
     then
-        export wwinit_kmods="${wwinit_uri}kmods&compress=gz"; info "wwinit_kmods=${wwinit_kmods}"
+        export wwinit_kmods="${wwinit_uri}&stage=kmods&compress=gz"; info "wwinit_kmods=${wwinit_kmods}"
     fi
 
     wwinit_tmpfs_size=$(getarg wwinit.tmpfs.size=)