diff --git a/src/main/java/com/smart/generator/builder/impl/EntityBuilder.java b/src/main/java/com/smart/generator/builder/impl/EntityBuilder.java
index b1feb198de67d406dde3830def1acf91a9480d0c..581a808ac837fa27f840477d5838ae3ef59f5ccd 100644
--- a/src/main/java/com/smart/generator/builder/impl/EntityBuilder.java
+++ b/src/main/java/com/smart/generator/builder/impl/EntityBuilder.java
@@ -5,6 +5,7 @@ import com.smart.framework.util.VelocityUtil;
 import com.smart.generator.bean.Column;
 import com.smart.generator.bean.Field;
 import com.smart.generator.bean.Table;
+import com.smart.generator.builder.Builder;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.HashMap;
diff --git a/src/main/java/com/smart/generator/builder/impl/SQLBuilder.java b/src/main/java/com/smart/generator/builder/impl/SQLBuilder.java
index fe55998b08fc508bb2688d5d3d20cc8e41110100..4b3ef606ad8488a76d52c5288a5ef687a1323d69 100644
--- a/src/main/java/com/smart/generator/builder/impl/SQLBuilder.java
+++ b/src/main/java/com/smart/generator/builder/impl/SQLBuilder.java
@@ -3,6 +3,7 @@ package com.smart.generator.builder.impl;
 import com.smart.framework.util.VelocityUtil;
 import com.smart.generator.bean.Column;
 import com.smart.generator.bean.Table;
+import com.smart.generator.builder.Builder;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;