diff --git a/drive.inc b/drive.inc
index e06a00c..2e4bb93 100644
--- a/drive.inc
+++ b/drive.inc
@@ -13,6 +13,9 @@ function setup_drive
         mkpart '"root unused"' ext4 2049 3072 \
         mkpart '"root unused"' ext4 3073 4096 \
         mkpart Encrypted ext4 4097 -1
+    until [ -b "$drive"1 ] || [ -b "$drive"p1 ]; do
+        sleep 1
+    done
 } 
 
 function rename_partition