diff --git a/gentoo-gnome.sh b/gentoo-gnome.sh
index 123bf09..9ea6373 100644
--- a/gentoo-gnome.sh
+++ b/gentoo-gnome.sh
@@ -464,14 +464,13 @@ btrfs-tweaks() {
 
 initialize-disk() {
 	SWAP_SIZE=$(expr $(cat /proc/meminfo | head -1 | cut -d ':' -f2 | sed 's/ //g' | sed 's/kB//') / 1024)
+	SWAP_SIZE_END=$(expr $SWAP_SIZE + $ROOT_PARTITION_SIZE)
 	echo "Test: $SWAP_SIZE"
 	parted /dev/$HARD_DISK mklabel gpt
 	parted -a optimal /dev/$HARD_DISK mkpart primary fat32 1MiB 200MiB
 	parted -a optimal /dev/$HARD_DISK mkpart root ext3 200MiB $ROOT_PARTITION_SIZE
 	parted -a optimal /dev/$HARD_DISK set 1 esp on
-	echo $(cat /proc/meminfo | head -1 | cut -d ':' -f2 | sed 's/ //g')
-	sleep 5
-	parted -a optimal /dev/$HARD_DISK mkpart swap ext3 $ROOT_PARTITION_SIZE $SWAP_SIZE
+	parted -a optimal /dev/$HARD_DISK mkpart swap ext3 $ROOT_PARTITION_SIZE $SWAP_SIZE_END
 	parted -a optimal /dev/$HARD_DISK mkpart home ext3 $SWAP_SIZE 100%
 	partitionDetection
 	echo