diff --git a/vps b/vps
index 0115c03..ef54566 100755
--- a/vps
+++ b/vps
@@ -78,7 +78,7 @@ setup_chroot() {
 	echo "tmpfs           /tmp        tmpfs   defaults,nosuid,nodev   0 0" >> /etc/fstab
 
 	touch /swapfile
-	dd if=/dev/zero of=/swapfile bs=1M count=$SWAP_SIZE status=progress
+	dd if=/dev/zero of=/swapfile bs=1M count=$(($SWAP_SIZE_GB * 1024)) status=progress
 	chmod 0600 /swapfile
 	mkswap /swapfile && swapon /swapfile
 	echo "/swapfile none swap sw 0 0" >> /etc/fstab