diff --git a/docs/deploy/e2b.mdx b/docs/deploy/e2b.mdx
index ed098f3..cfe1d66 100644
--- a/docs/deploy/e2b.mdx
+++ b/docs/deploy/e2b.mdx
@@ -26,6 +26,10 @@ await sandbox.commands.run(
   "curl -fsSL https://releases.rivet.dev/sandbox-agent/latest/install.sh | sh"
 );
 
+// Install agents before starting the server
+await sandbox.commands.run("sandbox-agent install-agent claude");
+await sandbox.commands.run("sandbox-agent install-agent codex");
+
 // Start the server in the background
 await sandbox.commands.run(
   "sandbox-agent server --no-token --host 0.0.0.0 --port 3000",
@@ -46,10 +50,6 @@ for (let i = 0; i < 30; i++) {
   }
 }
 
-// Install agents (or pre-install in a custom template)
-await client.installAgent("claude");
-await client.installAgent("codex");
-
 // Create a session and start coding
 await client.createSession("my-session", {
   agent: "claude",