diff --git a/02-openwebui/fleet.yaml b/02-openwebui/fleet.yaml
index 3bef622..448619f 100644
--- a/02-openwebui/fleet.yaml
+++ b/02-openwebui/fleet.yaml
@@ -59,21 +59,6 @@ helm:
                 - "*"
               maxAge: 3600
               allowCredentials: true
-      additionalRules:
-        - backendRefs:
-          - filters:
-            - type: CORS
-              cors:
-                 allowOrigins: 
-                   - "*"
-                 allowMethods: 
-                   - "*"
-                 allowHeaders: 
-                   - "*"
-                 exposeHeaders: 
-                   - "*"
-                 maxAge: 3600
-                 allowCredentials: true
              
 
     extraEnvVars:
diff --git a/04-agentgateway-proxy/templates/trafficpolicy.yaml b/04-agentgateway-proxy/templates/trafficpolicy.yaml
new file mode 100644
index 0000000..229076c
--- /dev/null
+++ b/04-agentgateway-proxy/templates/trafficpolicy.yaml
@@ -0,0 +1,13 @@
+apiVersion: gateway.kgateway.dev/v1alpha1
+kind: HTTPListenerPolicy
+metadata:
+  name: websocket-proxy
+spec:
+  targetRefs:
+    - name: http
+      kind: Gateway
+      group: gateway.networking.k8s.io
+  upgradeConfig:
+    enabledUpgrades:
+      - websocket
+    
\ No newline at end of file