diff --git a/nixos-infect b/nixos-infect
index 88e8305..08b085c 100755
--- a/nixos-infect
+++ b/nixos-infect
@@ -258,4 +258,4 @@ makeConf
 makeSwap # smallest (512MB) droplet needs extra memory!
 infect
 removeSwap
-reboot
+[[ -z "$NO_REBOOT" ]] && reboot