diff --git a/src/init.sh b/src/init.sh
index c9313a4..63582d8 100755
--- a/src/init.sh
+++ b/src/init.sh
@@ -4,4 +4,4 @@
 #src/iso.sh
 
 printf '%b\n' "\n\033[1m"WINDOWS_INIT:"\033[0m"" Executing vm.sh"
-sudo src/vm.sh
+sudo src/vm.sh /dev/sdb     # change to needed drive
diff --git a/src/vm.sh b/src/vm.sh
index b6f00fe..85a9520 100755
--- a/src/vm.sh
+++ b/src/vm.sh
@@ -6,7 +6,7 @@
 # libvirt-daemon for /shrug
 SPICE_PORT=5930
 CHARDEV_ID='spicechannel0'
-DRIVE='/dev/sdb'    # change drive if needed
+DRIVE=$argv
 ISO='Win*.iso'
 sudo qemu-system-x86_64 \
     -enable-kvm \
@@ -18,12 +18,13 @@ sudo qemu-system-x86_64 \
     -m 8G \
     -boot d \
     -cdrom ${ISO} \
-    -drive file=${DRIVE},format=raw,cache=none,if=virtio \
+    -hda ${DRIVE} \
     -vga qxl \
     -device virtio-serial-pci \
     -spice port=${SPICE_PORT},disable-ticketing=on \
     -device virtserialport,chardev=${CHARDEV_ID},name=com.redhat.spice.0 \
     -chardev spicevmc,id=${CHARDEV_ID},name=vdagent
+    #-drive file=${DRIVE},format=raw,cache=none,if=virtio \
 
 exec spicy --title Windows 127.0.0.1 -p ${SPICE_PORT}