diff --git a/flake.lock b/flake.lock
index 01b863e..b06d5c7 100644
--- a/flake.lock
+++ b/flake.lock
@@ -168,11 +168,11 @@
     "wallpaper-engine-plasma6-plugin": {
       "flake": false,
       "locked": {
-        "lastModified": 1710304679,
-        "narHash": "sha256-vkWEGlDQpfJ3fAimJHZs+aX6dh/fLHSRy2tLEsgu/JU=",
+        "lastModified": 1718553568,
+        "narHash": "sha256-bKGQxyS8gUi+37lODLVHphMoQwLKZt/hpSjR5MN+5GA=",
         "ref": "qt6",
-        "rev": "96230de92f1715d3ccc5b9d50906e6a73812a00a",
-        "revCount": 604,
+        "rev": "1e604105c586c7938c5b2c19e3dc8677b2eb4bb4",
+        "revCount": 608,
         "submodules": true,
         "type": "git",
         "url": "https://github.com/catsout/wallpaper-engine-kde-plugin"
diff --git a/packages/wallpaper-engine-plasma5-plugin.nix b/packages/wallpaper-engine-plasma5-plugin.nix
index 4c91dc6..5b03d86 100644
--- a/packages/wallpaper-engine-plasma5-plugin.nix
+++ b/packages/wallpaper-engine-plasma5-plugin.nix
@@ -42,6 +42,7 @@ pkgs.plasma5Packages.callPackage
       postInstall = ''
         chmod +x ../plugin/contents/pyext.py
         PATH=${python-ws}/bin:$PATH patchShebangs --build ../plugin/contents/pyext.py
+        sed -i "18i \\            \"[ -f \\\\\"${placeholder "out"}/share/\$EXT\\\\\" ] && WKD=\\\\\"${placeholder "out"}/share/\$EXT\\\\\"\"," ../plugin/contents/ui/Pyext.qml
         sed -i "s/exec python3 //g" ../plugin/contents/ui/Pyext.qml
         mkdir -p $out/share/plasma/wallpapers/com.github.casout.wallpaperEngineKde/
         cp -r ../plugin/* $out/share/plasma/wallpapers/com.github.casout.wallpaperEngineKde/
diff --git a/packages/wallpaper-engine-plasma6-plugin.nix b/packages/wallpaper-engine-plasma6-plugin.nix
index 7a21b7b..a1afa1a 100644
--- a/packages/wallpaper-engine-plasma6-plugin.nix
+++ b/packages/wallpaper-engine-plasma6-plugin.nix
@@ -30,9 +30,10 @@ pkgs.kdePackages.callPackage
       meta.license = with lib.licenses; [ gpl2 ];
       pname = name;
       postInstall = ''
-        sed -i "s/exec python3 //g" ../plugin/contents/ui/Pyext.qml
         chmod +x ../plugin/contents/pyext.py
         PATH=${python-ws}/bin:$PATH patchShebangs --build ../plugin/contents/pyext.py
+        sed -i "18i \\            \"[ -f \\\\\"${placeholder "out"}/share/\$EXT\\\\\" ] && WKD=\\\\\"${placeholder "out"}/share/\$EXT\\\\\"\"," ../plugin/contents/ui/Pyext.qml
+        sed -i "s/exec python3 //g" ../plugin/contents/ui/Pyext.qml
         kpackagetool6 -i ../plugin/ -p $out/share/plasma/wallpapers/
       '';
       src = inputs.${name};