diff --git a/mcp/windmill_mcp.py b/mcp/windmill_mcp.py
index b9f4eed..f14189f 100644
--- a/mcp/windmill_mcp.py
+++ b/mcp/windmill_mcp.py
@@ -10,12 +10,14 @@ from mcp.server.fastmcp import FastMCP
 WINDMILL_URL = os.environ.get("WINDMILL_URL", "https://windmill.keinafarm.net")
 WINDMILL_TOKEN = os.environ.get("WINDMILL_TOKEN", "")
 WINDMILL_WORKSPACE = os.environ.get("WINDMILL_WORKSPACE", "admins")
+MCP_HOST = os.environ.get("MCP_HOST", "127.0.0.1")
+MCP_PORT = int(os.environ.get("MCP_PORT", "8001"))
 
 if not WINDMILL_TOKEN:
     print("Error: WINDMILL_TOKEN 環境変数が設定されていません", file=sys.stderr)
     sys.exit(1)
 
-mcp = FastMCP("windmill")
+mcp = FastMCP("windmill", host=MCP_HOST, port=MCP_PORT)
 
 
 def _headers() -> dict:
@@ -338,9 +340,4 @@ def windmill_get_script(path: str) -> str:
 
 if __name__ == "__main__":
     transport = os.environ.get("MCP_TRANSPORT", "stdio")
-    if transport == "sse":
-        host = os.environ.get("MCP_HOST", "0.0.0.0")
-        port = int(os.environ.get("MCP_PORT", "8001"))
-        mcp.run(transport="sse", host=host, port=port)
-    else:
-        mcp.run(transport="stdio")
+    mcp.run(transport=transport)