diff --git a/post-debootstrap-installer.sh b/post-debootstrap-installer.sh
index 6f862cc..0d4bc15 100755
--- a/post-debootstrap-installer.sh
+++ b/post-debootstrap-installer.sh
@@ -27,7 +27,7 @@ cat /hardware-setup/authorized_keys/* > /etc/dropbear-initramfs/authorized_keys
 apt-get -y install linux-image-amd64
 
 DEBIAN_FRONTEND=noninteractive apt-get -y install grub-pc
-/hardware-setup/grub.sh
+"/hardware-setup/hardware/$1/grub.sh"
 update-grub
 
 ### users
diff --git a/setup.sh b/setup.sh
index 9c26c04..d80eca0 100755
--- a/setup.sh
+++ b/setup.sh
@@ -48,6 +48,6 @@ echo "$2" > "$chroot/etc/hostname"
 mkdir "$chroot/hardware-setup"
 cp -a * "$chroot/hardware-setup"
 
-chroot "$chroot" /hardware-setup/post-debootstrap-installer.sh
+chroot "$chroot" /hardware-setup/post-debootstrap-installer.sh "$1"
 
 rm -r "$chroot/hardware-setup"