diff --git a/src/lib/onboard.ts b/src/lib/onboard.ts
index 5dbb26affc0..ebfc028d857 100644
--- a/src/lib/onboard.ts
+++ b/src/lib/onboard.ts
@@ -5251,6 +5251,7 @@ async function onboard(opts = {}) {
         agent,
         dangerouslySkipPermissions,
       );
+      registry.updateSandbox(sandboxName, { model, provider });
       onboardSession.markStepComplete("sandbox", { sandboxName, provider, model, nimContainer });
     }