diff --git a/black_forest_streamlit.py b/black_forest_streamlit.py
index 99c7287..b7856cf 100644
--- a/black_forest_streamlit.py
+++ b/black_forest_streamlit.py
@@ -189,7 +189,7 @@ def poll_for_result(conn, headers, request_id):
 
 def generate_image(prompt, api_key, endpoint, seed):
     """Generate an image using the Black Forest Labs API."""
-    conn = http.client.HTTPSConnection("api.bfl.ml")
+    conn = http.client.HTTPSConnection("api.bfl.ai")
     headers = {"Content-Type": "application/json", "X-Key": api_key}
     payload = {
         "prompt": prompt,
diff --git a/black_forest_test.py b/black_forest_test.py
index 5f2cc9e..ac9cce4 100644
--- a/black_forest_test.py
+++ b/black_forest_test.py
@@ -51,7 +51,7 @@ def main():
     if not api_key:
         raise ValueError("BLACK_FOREST_LABS_API_KEY environment variable is not set")
 
-    conn = http.client.HTTPSConnection("api.bfl.ml")
+    conn = http.client.HTTPSConnection("api.bfl.ai")
     headers = {"Content-Type": "application/json", "X-Key": api_key}
 
     payload = {