diff --git a/examples/websocket-service/start_web_server.py b/examples/websocket-service/start_web_server.py
index 1ecb1b6..b794edb 100755
--- a/examples/websocket-service/start_web_server.py
+++ b/examples/websocket-service/start_web_server.py
@@ -40,9 +40,8 @@ def signal_handler(sig, frame):
     """Handle Ctrl+C gracefully."""
     print("\n👋 Shutting down...")
     if httpd:
+        # Use shutdown() to stop serve_forever()
         httpd.shutdown()
-        httpd.server_close()
-    sys.exit(0)
 
 
 def main():
@@ -75,15 +74,16 @@ def main():
 ╚══════════════════════════════════════════════════════════╝
     """)
 
-    with socketserver.TCPServer((args.host, args.port), CustomHandler) as httpd:
-        print(f"🚀 Server started on http://{args.host}:{args.port}")
-        try:
-            httpd.serve_forever()
-        except KeyboardInterrupt:
-            pass
-        finally:
-            print("\n👋 Server stopped")
-            httpd.server_close()
+    httpd = socketserver.TCPServer((args.host, args.port), CustomHandler)
+    print(f"🚀 Server started on http://{args.host}:{args.port}")
+    
+    try:
+        httpd.serve_forever()
+    except KeyboardInterrupt:
+        pass
+    finally:
+        print("\n👋 Server stopped")
+        httpd.server_close()
 
 
 if __name__ == "__main__":