diff --git a/nixos-infect b/nixos-infect
index 70b9bc8..1bcf646 100755
--- a/nixos-infect
+++ b/nixos-infect
@@ -127,11 +127,8 @@ makeSwap() {
 }
 
 removeSwap() {
-    for swapFile in /tmp/nixos-infect.*.swp
-    do
-        swapoff -v "$swapFile"
-        rm -vf "$swapFile"
-    done
+    swapoff -a
+    rm -vf /tmp/nixos-infect.*.swp
 }
 
 prepareEnv() {