diff --git a/agent/src/agent.rs b/agent/src/agent.rs
index c7a3064..652d1b1 100644
--- a/agent/src/agent.rs
+++ b/agent/src/agent.rs
@@ -304,6 +304,8 @@ impl Agent {
                 .arg("cm")
                 .arg("/run/current-system/sw/bin/nixos-rebuild")
                 .arg("switch")
+                .arg("--flake")
+                .arg(".")
                 .current_dir(working_dir)
                 .output()
                 .await