diff --git a/src/main/java/cn/rainss/codegenerator/utils/GeneratorUtil.java b/src/main/java/cn/rainss/codegenerator/utils/GeneratorUtil.java
index 57a97bc..235d2f2 100644
--- a/src/main/java/cn/rainss/codegenerator/utils/GeneratorUtil.java
+++ b/src/main/java/cn/rainss/codegenerator/utils/GeneratorUtil.java
@@ -206,7 +206,7 @@ public class GeneratorUtil {
             String javaType = column.getJavaType();
             if (DataType.DECIMAL.getJavaType().equals(javaType)) {
                 tableInfo.setHasBigDecimal(Boolean.TRUE);
-            } else if (DataType.TIMESTAMP.getSqlType().equals(javaType)) {
+            } else if (DataType.TIMESTAMP.getJavaType().equals(javaType)) {
                 tableInfo.setHasLocalDateTime(Boolean.TRUE);
             } else if (DataType.DATE.getJavaType().equals(javaType)) {
                 tableInfo.setHasLocalDate(Boolean.TRUE);