fixed swap problem
diff --git a/ais b/ais
index 1dccaff..0564c7d 100755
--- a/ais
+++ b/ais
@@ -158,7 +158,7 @@
   }
   umount_partition(){
     #umount swaps
-    swapoff -a
+    swapon -s|grep $1 && swapoff $1
     #check if partition is mounted and umount
     mount|grep $1 && umount $1
   }
@@ -178,7 +178,7 @@
       fsck $1
       mkdir -p $2
       mount -t $TYPE $1 $2
-      remove_partition "$1"
+      remove_partition
     fi
   }
   format_swap_partition(){
@@ -187,7 +187,7 @@
       umount_partition "$1"
       mkswap $1
       swapon $1
-      remove_partition "$1"
+      remove_partition
     fi
   }
   create_swap(){