diff --git a/pom.xml b/pom.xml
index 51c822e..0b49d1d 100644
--- a/pom.xml
+++ b/pom.xml
@@ -3,7 +3,7 @@
     <modelVersion>4.0.0</modelVersion>
     <groupId>com.ethlo.persistence.tools</groupId>
     <artifactId>eclipselink-maven-plugin</artifactId>
-    <version>3.0.1</version>
+    <version>3.0.2</version>
     <packaging>maven-plugin</packaging>
     <name>Eclipselink Maven Plugin</name>
     <description>Maven Plugin for Eclipselink supporting static weaving, meta-model generation and DDL generation
diff --git a/src/main/java/com/ethlo/persistence/tools/eclipselink/EclipselinkModelGenMojo.java b/src/main/java/com/ethlo/persistence/tools/eclipselink/EclipselinkModelGenMojo.java
index f18c5f4..8666a36 100644
--- a/src/main/java/com/ethlo/persistence/tools/eclipselink/EclipselinkModelGenMojo.java
+++ b/src/main/java/com/ethlo/persistence/tools/eclipselink/EclipselinkModelGenMojo.java
@@ -193,7 +193,17 @@ public void execute() throws MojoExecutionException
 
     private Set<File> getSourceFiles()
     {
-        if (source == null || !source.exists())
+        final Set<File> fromSource = getFilesFromDirectory(source);
+        final Set<File> fromGenerated = getFilesFromDirectory(generatedSourcesDirectory);
+        final Set<File> all = new TreeSet<>();
+        all.addAll(fromSource);
+        all.addAll(fromGenerated);
+        return all;
+    }
+
+    private Set<File> getFilesFromDirectory(File dir)
+    {
+        if (dir == null || !dir.exists())
         {
             return new TreeSet<>();
         }
@@ -208,12 +218,12 @@ private Set<File> getSourceFiles()
             }
         }
 
-        Set<File> files = new HashSet<>();
-        final Scanner scanner = buildContext.newScanner(source);
+        final Set<File> files = new HashSet<>();
+        final Scanner scanner = buildContext.newScanner(dir);
         scanner.setIncludes(filters);
         scanner.scan();
 
-        String[] includedFiles = scanner.getIncludedFiles();
+        final String[] includedFiles = scanner.getIncludedFiles();
         if (includedFiles != null)
         {
             for (String includedFile : includedFiles)