diff --git a/xiaonuo-base/xiaonuo-gen/src/main/java/com/cn/xiaonuo/generate/core/util/Util.java b/xiaonuo-base/xiaonuo-gen/src/main/java/com/cn/xiaonuo/generate/core/util/Util.java
index 77d88a97..0f8b920b 100644
--- a/xiaonuo-base/xiaonuo-gen/src/main/java/com/cn/xiaonuo/generate/core/util/Util.java
+++ b/xiaonuo-base/xiaonuo-gen/src/main/java/com/cn/xiaonuo/generate/core/util/Util.java
@@ -83,7 +83,7 @@ public class Util {
     public static String getDataBasename () {
         String dataUrl = ConstantContext.me().getStr(CommonConstant.DATABASE_URL_NAME);
         String driverName = ConstantContext.me().getStr(CommonConstant.DATABASE_DRIVER_NAME);
-        if (driverName.contains(DbIdEnum.MS_SQL.getCode())) {
+        if (driverName.contains(DbIdEnum.MYSQL.getCode())) {
             return dataUrl.substring(getIndex(dataUrl, 3, "/") + 1, dataUrl.indexOf("?"));
         } else if (driverName.contains(DbIdEnum.ORACLE.getCode())) {
             return ConstantContext.me().getStr(CommonConstant.DATABASE_USER_NAME);