diff --git a/entry.sh b/entry.sh
index 6aeae4c..99d5a29 100755
--- a/entry.sh
+++ b/entry.sh
@@ -73,7 +73,7 @@ function first_stage(){
 	pvcreate ${DEVICE}2
 	vgcreate vg0 ${DEVICE}2
 	
-	export LVM_SIZE=$(dev_size /dev/mapper/cryptlvm)
+	export LVM_SIZE=$(dev_size ${DEVICE}2)
 	
 	export SWAP_SIZE=$(($(mem_size)/1024/1024))
 	export ROOT_SIZE=$(($LVM_SIZE/1024/1024-$SWAP_SIZE-512))