diff --git a/nixos-infect b/nixos-infect
index 9a199d4..21a2d06 100644
--- a/nixos-infect
+++ b/nixos-infect
@@ -444,9 +444,10 @@ if [[ -z "$NO_SWAP" ]]; then
 fi
 
 makeConf
-bootctl install --graceful # Add necessary boot files.
 infect
 
+bootctl install --graceful # Add necessary boot files.
+
 if [[ -z "$NO_SWAP" ]]; then
 	removeSwap
 fi