diff --git a/modules/base/initrd/default.nix b/modules/base/initrd/default.nix
index fa57911..ac27b33 100644
--- a/modules/base/initrd/default.nix
+++ b/modules/base/initrd/default.nix
@@ -26,6 +26,14 @@
         authorizedKeys = config.mj.base.initrd.authorizedKeys;
         hostKeys = config.mj.base.initrd.hostKeys;
       };
+      postCommands = ''
+        tee -a /root/.profile >/dev/null <<EOF
+        if zfs load-key rpool/nixos; then
+           pkill zfs
+        fi
+        exit
+        EOF'';
+    };
     };
   };
 }