diff --git a/main.py b/main.py
index 8fb97c5..b11ef6a 100644
--- a/main.py
+++ b/main.py
@@ -517,7 +517,12 @@ def main():
                 )
                 sys.exit(1)
 
-            server.run(transport="streamable-http", host=host, port=port)
+            server.run(
+                transport="streamable-http",
+                host=host,
+                port=port,
+                stateless_http=is_stateless_mode(),
+            )
         else:
             server.run()
     except KeyboardInterrupt: