diff --git a/chat2db-server/chat2db-spi/src/main/java/ai/chat2db/spi/sql/Chat2DBContext.java b/chat2db-server/chat2db-spi/src/main/java/ai/chat2db/spi/sql/Chat2DBContext.java
index 246031c8..b44344f6 100644
--- a/chat2db-server/chat2db-spi/src/main/java/ai/chat2db/spi/sql/Chat2DBContext.java
+++ b/chat2db-server/chat2db-spi/src/main/java/ai/chat2db/spi/sql/Chat2DBContext.java
@@ -99,6 +99,8 @@ public class Chat2DBContext {
         String host = connectInfo.getHost();
         String port = connectInfo.getPort() + "";
         try {
+            ssh.setRHost(host);
+            ssh.setRPort(port);
             session = getSession(ssh);
             if (session != null) {
                 url = url.replace(host, "127.0.0.1").replace(port, ssh.getLocalPort());