diff --git a/examples/mcp_example/calc_server.py b/examples/mcp_example/calc_server.py
index 797e75af..5433140b 100644
--- a/examples/mcp_example/calc_server.py
+++ b/examples/mcp_example/calc_server.py
@@ -33,4 +33,4 @@ def break_even_point(fixed_costs: float, price_per_unit: float, cost_per_unit: f
 
 if __name__ == "__main__":
     print("Starting Business Calculator Server on port 6275...")
-    mcp.run(transport="sse", transport_kwargs={"host": "0.0.0.0", "port": 6275})
+    mcp.run(transport="sse", host="0.0.0.0", port=6275)
diff --git a/examples/mcp_example/math_server.py b/examples/mcp_example/math_server.py
index 70221308..ca7eaf2e 100644
--- a/examples/mcp_example/math_server.py
+++ b/examples/mcp_example/math_server.py
@@ -36,4 +36,4 @@ def calculate_percentage(part: float, whole: float) -> float:
 
 if __name__ == "__main__":
     print("Starting Math Server on port 6274...")
-    mcp.run(transport="sse", transport_kwargs={"host": "0.0.0.0", "port": 6274})
+    mcp.run(transport="sse", host="0.0.0.0", port=6274)
diff --git a/examples/mcp_example/test_integration.py b/examples/mcp_example/test_integration.py
index d49e9887..102ad828 100644
--- a/examples/mcp_example/test_integration.py
+++ b/examples/mcp_example/test_integration.py
@@ -54,7 +54,10 @@ def main():
     
     while True:
         try:
-            user_input = input("\nEnter your request: ")
+            user_input = input("\nEnter your request (or Ctrl+C to exit): ")
+        except KeyboardInterrupt:
+            print("\nExiting gracefully...")
+            break
             
             if user_input.lower() == 'exit':
                 break