diff --git a/modules/outputs/initramfs.nix b/modules/outputs/initramfs.nix
index b1e0ea8..6d5d473 100644
--- a/modules/outputs/initramfs.nix
+++ b/modules/outputs/initramfs.nix
@@ -54,7 +54,7 @@ in
             mount -t sysfs none /sys
             ${busybox}/bin/sh
           '';
-          refs = pkgs.writeReferencesToFile busybox;
+          refs = pkgs.writeClosure [ busybox ];
         in runCommand "initramfs.cpio" {} ''
           cat << SPECIALS | ${gen_init_cpio}/bin/gen_init_cpio /dev/stdin > out
           dir /proc 0755 0 0
diff --git a/pkgs/initramfs-peek/default.nix b/pkgs/initramfs-peek/default.nix
index 39067cb..86f8e18 100644
--- a/pkgs/initramfs-peek/default.nix
+++ b/pkgs/initramfs-peek/default.nix
@@ -3,7 +3,7 @@
 , pkgsBuildBuild
 , runCommand
 , cpio
-, writeReferencesToFile
+, writeClosure
 , writeScript
 } :
 let
@@ -18,7 +18,7 @@ let
     mount -t sysfs none /sys
     ${busybox}/bin/sh
   '';
-  refs = writeReferencesToFile busybox;
+  refs = writeClosure [ busybox ];
 in runCommand "initramfs.cpio" { } ''
   cat << SPECIALS | ${gen_init_cpio}/bin/gen_init_cpio /dev/stdin > out
   dir /proc 0755 0 0