diff --git a/mybatis-plus-ext-processor/src/main/java/com/tangzc/mpe/processer/builder/MapperBuilder.java b/mybatis-plus-ext-processor/src/main/java/com/tangzc/mpe/processer/builder/MapperBuilder.java
index 9423603d34cff556b96dd1e9b607cf79547ae95b..d4d52ee1e3c08aca90edefd89cd3dd43d16484cb 100644
--- a/mybatis-plus-ext-processor/src/main/java/com/tangzc/mpe/processer/builder/MapperBuilder.java
+++ b/mybatis-plus-ext-processor/src/main/java/com/tangzc/mpe/processer/builder/MapperBuilder.java
@@ -59,7 +59,7 @@ public class MapperBuilder extends BaseBuilder {
         List<String> lines = Arrays.asList(
                 "package " + mapperPackageName + ";",
                 "",
-                "import com.baomidou.mybatisplus.core.mapper.BaseMapper;",
+                "import " + mapperSuperclassName.canonicalName() + ";",
                 dsAnnoImport,
                 "import " + entityPackageName + "." + entityName + ";",
                 "import org.apache.ibatis.annotations.Mapper;",