diff --git a/config.py b/config.py
index 2b9bd03..542613e 100644
--- a/config.py
+++ b/config.py
@@ -2,8 +2,7 @@
 import functools
 import os
 
-from fal_client.auth import MissingCredentialsError
-
+from fal_client.auth import MissingCredentialsError, FAL_RUN_HOST
 
 @functools.cache
 def get_fal_config():
@@ -18,7 +17,7 @@ def get_fal_config():
 def get_fal_endpoint():
     config = get_fal_config()
     endpoint = config["fal"]["application_name"]
-    return f"https://fal.run/{endpoint}"
+    return f"https://{FAL_RUN_HOST}/{endpoint}"
 
 
 @functools.cache
diff --git a/pyproject.toml b/pyproject.toml
index dc15ed3..7eacfd6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-fal-connector"
 description = "The ComfyUI-fal-Connector is a tool designed to provide an integration between ComfyUI and fal. This extension allows users to execute their ComfyUI workflows directly on [a/fal.ai](https://fal.ai/). This enables users to leverage the computational power and resources provided by fal.ai for running their ComfyUI workflows."
-version = "1.0.0"
+version = "1.0.1"
 license = { file = "LICENSE" }
 dependencies = ["fal-client>=0.3.0", "httpx", "httpx-sse", "fal"]