diff --git a/setup.sh b/setup.sh
index df5d0a6..6effed1 100644
--- a/setup.sh
+++ b/setup.sh
@@ -501,6 +501,7 @@ function main () {
             set_hostname
             set_locale
             add_zfs_hook_to_initramfs
+            add_zfs_files_to_new_os
             enter_chroot
             # We're done in chroot
             finalize_os_setup