diff --git a/packages/dbgpt-ext/src/dbgpt_ext/datasource/rdbms/dialect/starrocks/sqlalchemy/__init__.py b/packages/dbgpt-ext/src/dbgpt_ext/datasource/rdbms/dialect/starrocks/sqlalchemy/__init__.py
index 5515b3f80..4a0e193ba 100644
--- a/packages/dbgpt-ext/src/dbgpt_ext/datasource/rdbms/dialect/starrocks/sqlalchemy/__init__.py
+++ b/packages/dbgpt-ext/src/dbgpt_ext/datasource/rdbms/dialect/starrocks/sqlalchemy/__init__.py
@@ -18,6 +18,6 @@ from sqlalchemy.dialects import registry
 
 registry.register(
     "starrocks",
-    "dbgpt.datasource.rdbms.dialect.starrocks.sqlalchemy.dialect",
+    "dbgpt_ext.datasource.rdbms.dialect.starrocks.sqlalchemy.dialect",
     "StarRocksDialect",
 )
diff --git a/packages/dbgpt-ext/src/dbgpt_ext/datasource/rdbms/dialect/starrocks/sqlalchemy/dialect.py b/packages/dbgpt-ext/src/dbgpt_ext/datasource/rdbms/dialect/starrocks/sqlalchemy/dialect.py
index 766ce75b2..364bc65d2 100644
--- a/packages/dbgpt-ext/src/dbgpt_ext/datasource/rdbms/dialect/starrocks/sqlalchemy/dialect.py
+++ b/packages/dbgpt-ext/src/dbgpt_ext/datasource/rdbms/dialect/starrocks/sqlalchemy/dialect.py
@@ -20,7 +20,7 @@ from sqlalchemy import exc, log, text
 from sqlalchemy.dialects.mysql.pymysql import MySQLDialect_pymysql
 from sqlalchemy.engine import Connection
 
-from dbgpt.datasource.rdbms.dialect.starrocks.sqlalchemy import datatype
+from dbgpt_ext.datasource.rdbms.dialect.starrocks.sqlalchemy import datatype
 
 logger = logging.getLogger(__name__)