diff --git a/create-linux-swapfile.sh b/create-linux-swapfile.sh
index c850ee6..27cbc6c 100644
--- a/create-linux-swapfile.sh
+++ b/create-linux-swapfile.sh
@@ -6,9 +6,9 @@ if grep -q "$FSTAB_SWAP_ENTRY" "$FSTAB_FILE"; then
 	echo "Skipping creation of swap partion because entry allready exists in \"$FSTAB_FILE\"!"
 else
 	echo "Creating swap partition..." &&
-	sudo fallocate -l 16G "$SWAP_FILE" &&
+	sudo fallocate -l "$1" "$SWAP_FILE" &&
 	sudo chmod 600 "$SWAP_FILE" &&
 	sudo mkswap "$SWAP_FILE" &&
 	sudo swapon "$SWAP_FILE" &&
-	sudo sh -c "echo \"$FSTAB_SWAP_ENTRY\">>\"$FSTAB_FILE\""
+	sudo sh -c "echo \"$FSTAB_SWAP_ENTRY\">>\"$FSTAB_FILE\"" || exit 1
 fi