diff --git a/nixos-infect b/nixos-infect
index 06f0171..31c0ad6 100644
--- a/nixos-infect
+++ b/nixos-infect
@@ -399,7 +399,9 @@ if [[ -z "$NO_SWAP" ]]; then
     makeSwap # smallest (512MB) droplet needs extra memory!
 fi
 makeConf
-infect
+if [[ -z "$NO_INFECT" ]]; then
+    infect
+fi
 if [[ -z "$NO_SWAP" ]]; then
     removeSwap
 fi