diff --git a/hardware/filesystem.nix b/hardware/filesystem.nix
index 21f0fe7..fb19d06 100644
--- a/hardware/filesystem.nix
+++ b/hardware/filesystem.nix
@@ -11,8 +11,9 @@ in
     boot.initrd.systemd.services.rollback-root = {
         description = "Rollback btrfs root subvolume";
         wantedBy = [ "initrd.target" ];
-        after = [ "systemd-cryptsetup@.service" ];
+        after = [ "initrd-root-device.target" ];
         before = [ "sysroot.mount" ];
+        requires = [ "initrd-root-device.target" ];
         unitConfig.DefaultDependencies = "no";
         serviceConfig.Type = "oneshot";
         script = ''