fixes
diff --git a/ais b/ais
index 24231f3..abec990 100755
--- a/ais
+++ b/ais
@@ -324,6 +324,7 @@
     [[ -n $OPT ]] && break
   done
   echo "Review your fstab"
+  [[ -d $MOUNTPOINT/swapfile ]] && sed -i 's/\'${MOUNTPOINT}'//' $MOUNTPOINT/etc/fstab
   pause_function
   $EDITOR $MOUNTPOINT/etc/fstab
 }