diff --git a/packages/connector-ui/wrangler.toml b/packages/connector-ui/wrangler.toml
index 81129bf..343f5ba 100644
--- a/packages/connector-ui/wrangler.toml
+++ b/packages/connector-ui/wrangler.toml
@@ -10,6 +10,8 @@ main = "worker.mjs"
 [assets]
 directory = "./dist"
 binding = "ASSETS"
+not_found_handling = "single-page-application"
+run_worker_first = ["/api/*"]
 
 [[durable_objects.bindings]]
 name = "SESSION_RELAY"
@@ -26,9 +28,25 @@ routes = [
   { pattern = "agentconnect.staging.polygon.technology", custom_domain = true }
 ]
 
+[[env.staging.durable_objects.bindings]]
+name = "SESSION_RELAY"
+class_name = "SessionRelay"
+
+[[env.staging.migrations]]
+tag = "v1"
+new_classes = ["SessionRelay"]
+
 [env.production]
 name = "agentconnect-production"
 
 routes = [
   { pattern = "agentconnect.polygon.technology", custom_domain = true }
 ]
+
+[[env.production.durable_objects.bindings]]
+name = "SESSION_RELAY"
+class_name = "SessionRelay"
+
+[[env.production.migrations]]
+tag = "v1"
+new_classes = ["SessionRelay"]