diff --git a/src/main/java/com/ruoyi/common/utils/security/ShiroUtils.java b/src/main/java/com/ruoyi/common/utils/security/ShiroUtils.java
index 9bdf1ad0..02d5ca4a 100644
--- a/src/main/java/com/ruoyi/common/utils/security/ShiroUtils.java
+++ b/src/main/java/com/ruoyi/common/utils/security/ShiroUtils.java
@@ -6,7 +6,6 @@ import org.apache.shiro.session.Session;
 import org.apache.shiro.subject.PrincipalCollection;
 import org.apache.shiro.subject.SimplePrincipalCollection;
 import org.apache.shiro.subject.Subject;
-
 import com.ruoyi.framework.shiro.realm.UserRealm;
 import com.ruoyi.project.system.user.domain.User;
 
diff --git a/src/main/resources/templates/vm/java/domain.java.vm b/src/main/resources/templates/vm/java/domain.java.vm
index d798c039..c0735a61 100644
--- a/src/main/resources/templates/vm/java/domain.java.vm
+++ b/src/main/resources/templates/vm/java/domain.java.vm
@@ -1,8 +1,11 @@
 package ${package}.domain;
 
 import com.ruoyi.framework.web.domain.BaseEntity;
-#if(${hasBigDecimal})
-import java.math.BigDecimal;
+#foreach ($column in $columns)
+#if($column.attrType == 'Date')
+import java.util.Date;
+#break
+#end
 #end
 
 /**