diff --git a/nginx/nginx.conf b/nginx/nginx.conf
index 65063ffa..dd338542 100644
--- a/nginx/nginx.conf
+++ b/nginx/nginx.conf
@@ -94,7 +94,8 @@ http {
 
         # AppFlowy AI
         location /ai {
-            proxy_pass http://ai:5001;
+            set $ai ai;
+            proxy_pass http://$ai:5001;
             proxy_set_header Host $host;
             proxy_pass_request_headers on;
         }