diff --git a/entry.sh b/entry.sh
index 7db7457..b5577f5 100755
--- a/entry.sh
+++ b/entry.sh
@@ -113,7 +113,7 @@ function second_stage(){
 	sed -i 's/main/main contrib non-free/g' /etc/apt/sources.list
 	apt update
 	apt install -y linux-image-amd64 grub2 cryptsetup btrfs-progs lvm2 firmware-iwlwifi locales tzdata keyboard-configuration console-common zsh
-	grub-install 
+	grub-install ${DEVICE}
 	mkdir /snap
 	btrfs subvolume snapshot / /snap/$(date +%Y-%m-%d_basesystem)
 	apt install -y task-mate-desktop mate-desktop-environment-extra