diff --git a/gentoo-gnome.sh b/gentoo-gnome.sh
index edcdc6a..f72c4e2 100644
--- a/gentoo-gnome.sh
+++ b/gentoo-gnome.sh
@@ -470,8 +470,8 @@ initialize-disk() {
 	parted -a optimal /dev/$HARD_DISK mkpart primary fat32 1MiB 200MiB
 	parted -a optimal /dev/$HARD_DISK mkpart root ext3 200MiB ${ROOT_PARTITION_SIZE}GB
 	parted -a optimal /dev/$HARD_DISK set 1 esp on
-	parted -a optimal /dev/$HARD_DISK mkpart swap ext3 ${ROOT_PARTITION_SIZE}GB ${SWAP_SIZE_END}GB
-	parted -a optimal /dev/$HARD_DISK mkpart home ext3 ${SWAP_SIZE_END}GB 100%
+	parted -a optimal /dev/$HARD_DISK mkpart swap ext3 ${ROOT_PARTITION_SIZE}GB ${SWAP_SIZE_END}KB
+	parted -a optimal /dev/$HARD_DISK mkpart home ext3 ${SWAP_SIZE_END}KB 100%
 	sleep 10
 	partitionDetection
 	echo