diff --git a/pmd-swift/pom.xml b/pmd-swift/pom.xml
index f84fd9490f..4ca080ae2d 100644
--- a/pmd-swift/pom.xml
+++ b/pmd-swift/pom.xml
@@ -10,8 +10,33 @@
         <version>7.0.0-SNAPSHOT</version>
     </parent>
 
+    <properties>
+        <antlr4.visitor>true</antlr4.visitor>
+    </properties>
+
     <build>
         <plugins>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-antrun-plugin</artifactId>
+                <inherited>true</inherited>
+                <executions>
+                    <execution>
+                        <id>process-sources</id>
+                        <phase>process-sources</phase>
+                        <configuration>
+                            <target>
+                                <ant antfile="src/main/ant/antlr4.xml">
+                                    <property name="target" value="${project.build.directory}/generated-sources/antlr4" />
+                                </ant>
+                            </target>
+                        </configuration>
+                        <goals>
+                            <goal>run</goal>
+                        </goals>
+                    </execution>
+                </executions>
+            </plugin>
             <plugin>
                 <groupId>org.antlr</groupId>
                 <artifactId>antlr4-maven-plugin</artifactId>
diff --git a/pmd-swift/src/main/ant/antlr4.xml b/pmd-swift/src/main/ant/antlr4.xml
new file mode 100644
index 0000000000..60f6be9a3b
--- /dev/null
+++ b/pmd-swift/src/main/ant/antlr4.xml
@@ -0,0 +1,10 @@
+<project name="pmd" default="antlr4" basedir="../../../../">
+
+    <property name="target-package-dir" value="${target}/net/sourceforge/pmd/lang/swift/antlr4" />
+
+    <target name="antlr4" description="Generates all Antlr4 aspects within PMD">
+        <replace file="${target-package-dir}/SwiftParser.java"
+                 token="extends ParserRuleContext"
+                 value="extends net.sourceforge.pmd.lang.ast.AntlrBaseNode" />
+    </target>
+</project>
\ No newline at end of file