diff --git a/pmd-kotlin/src/main/java/net/sourceforge/pmd/lang/kotlin/AbstractKotlinRule.java b/pmd-kotlin/src/main/java/net/sourceforge/pmd/lang/kotlin/AbstractKotlinRule.java
index ac65ac86d6..31262b1473 100644
--- a/pmd-kotlin/src/main/java/net/sourceforge/pmd/lang/kotlin/AbstractKotlinRule.java
+++ b/pmd-kotlin/src/main/java/net/sourceforge/pmd/lang/kotlin/AbstractKotlinRule.java
@@ -5,8 +5,8 @@
 package net.sourceforge.pmd.lang.kotlin;
 
 import net.sourceforge.pmd.RuleContext;
-import net.sourceforge.pmd.lang.ast.AstVisitor;
 import net.sourceforge.pmd.lang.ast.impl.antlr4.AntlrBaseRule;
+import net.sourceforge.pmd.lang.kotlin.ast.KotlinVisitor;
 
 public abstract class AbstractKotlinRule extends AntlrBaseRule {
 
@@ -15,5 +15,5 @@ public abstract class AbstractKotlinRule extends AntlrBaseRule {
     }
 
     @Override
-    public abstract AstVisitor<RuleContext, ?> buildVisitor();
+    public abstract KotlinVisitor<RuleContext, ?> buildVisitor();
 }
diff --git a/pmd-kotlin/src/main/java/net/sourceforge/pmd/lang/kotlin/rule/errorprone/OverrideBothEqualsAndHashcodeRule.java b/pmd-kotlin/src/main/java/net/sourceforge/pmd/lang/kotlin/rule/errorprone/OverrideBothEqualsAndHashcodeRule.java
index 7a087c8ea9..de5f8a21be 100644
--- a/pmd-kotlin/src/main/java/net/sourceforge/pmd/lang/kotlin/rule/errorprone/OverrideBothEqualsAndHashcodeRule.java
+++ b/pmd-kotlin/src/main/java/net/sourceforge/pmd/lang/kotlin/rule/errorprone/OverrideBothEqualsAndHashcodeRule.java
@@ -8,7 +8,6 @@ package net.sourceforge.pmd.lang.kotlin.rule.errorprone;
 import java.util.List;
 
 import net.sourceforge.pmd.RuleContext;
-import net.sourceforge.pmd.lang.ast.AstVisitor;
 import net.sourceforge.pmd.lang.kotlin.AbstractKotlinRule;
 import net.sourceforge.pmd.lang.kotlin.ast.KotlinParser.KtClassDeclaration;
 import net.sourceforge.pmd.lang.kotlin.ast.KotlinParser.KtClassMemberDeclaration;
@@ -20,6 +19,7 @@ import net.sourceforge.pmd.lang.kotlin.ast.KotlinParser.KtFunctionValueParameter
 import net.sourceforge.pmd.lang.kotlin.ast.KotlinParser.KtModifiers;
 import net.sourceforge.pmd.lang.kotlin.ast.KotlinParser.KtSimpleIdentifier;
 import net.sourceforge.pmd.lang.kotlin.ast.KotlinTerminalNode;
+import net.sourceforge.pmd.lang.kotlin.ast.KotlinVisitor;
 import net.sourceforge.pmd.lang.kotlin.ast.KotlinVisitorBase;
 
 public class OverrideBothEqualsAndHashcodeRule extends AbstractKotlinRule {
@@ -27,7 +27,7 @@ public class OverrideBothEqualsAndHashcodeRule extends AbstractKotlinRule {
     private static final Visitor INSTANCE = new Visitor();
 
     @Override
-    public AstVisitor<RuleContext, ?> buildVisitor() {
+    public KotlinVisitor<RuleContext, ?> buildVisitor() {
         return INSTANCE;
     }