diff --git a/install.sh b/install.sh
index 9849d0e..e262e0c 100644
--- a/install.sh
+++ b/install.sh
@@ -17,7 +17,7 @@ cd /tmp
 curl https://gitlab.com/ahoneybun/nix-configs/-/raw/disko/partitions/simple-efi.nix -o /tmp/disko-config.nix
 
 # Replace drive in Disko file
-sudo sed -i "s#/dev/vdb#$rootName#g" /tmp/disko-config.nix
+sudo sed -i "s#/dev/sda#$driveName#g" /tmp/disko-config.nix
 
 # Run Disko to partition the disk
 sudo nix --experimental-features "nix-command flakes" run github:nix-community/disko -- --mode disko /tmp/disko-config.nix