diff --git a/setup.sh b/setup.sh
index 1ed05df..dddbc1a 100644
--- a/setup.sh
+++ b/setup.sh
@@ -24,6 +24,7 @@ for choice in $choices; do
         echo "输入Swap内存大小(MB) [默认: 2048]: "
         read -t 30 -p "> " swap_size
         if [ $? -ne 0 ] || [ -z "$swap_size" ]; then
+            swap_size=2048
             echo -e "\n使用默认值: ${swap_size}MB"
         fi
         dd if=/dev/zero of=/swapfile bs=1M count=$swap_size