diff --git a/src/main/java/com/cool/core/util/EntityUtils.java b/src/main/java/com/cool/core/util/EntityUtils.java
index 2657b04..6764611 100644
--- a/src/main/java/com/cool/core/util/EntityUtils.java
+++ b/src/main/java/com/cool/core/util/EntityUtils.java
@@ -28,10 +28,10 @@ public class EntityUtils {
         PathMatchingResourcePatternResolver resolver = new PathMatchingResourcePatternResolver();
         Resource[] resources = null;
         try {
-            resources = resolver.getResources("classpath*:com/cool/modules/**/entity/**/*.class");
+            resources = resolver.getResources("classpath*:com/cool/**/entity/*.class");
             for (Resource r : resources) {
                 String path = r.getURL().getPath();
-                String className = path.substring(path.indexOf("com/cool/modules"),
+                String className = path.substring(path.indexOf("com/cool"),
                     path.lastIndexOf('.')).replace('/', '.');
                 entitySet.add(className);
             }