diff --git a/src/main/java/org/jastadd/tooling/RelAstGrammarLineMarkerProvider.java b/src/main/java/org/jastadd/tooling/RelAstGrammarLineMarkerProvider.java
index a5ed75382b000c816c57b57d467b1673e41e7a62..1e8700833a57def53b05c01b1e6b3e19dddec361 100644
--- a/src/main/java/org/jastadd/tooling/RelAstGrammarLineMarkerProvider.java
+++ b/src/main/java/org/jastadd/tooling/RelAstGrammarLineMarkerProvider.java
@@ -5,6 +5,7 @@ import com.intellij.codeInsight.daemon.RelatedItemLineMarkerProvider;
 import com.intellij.codeInsight.navigation.NavigationGutterIconBuilder;
 import com.intellij.psi.PsiClass;
 import com.intellij.psi.PsiElement;
+import com.intellij.psi.PsiIdentifier;
 import org.jastadd.tooling.psi.RelAstGrammarTypeDecl;
 import org.jetbrains.annotations.NotNull;
 
@@ -20,7 +21,11 @@ public class RelAstGrammarLineMarkerProvider extends RelatedItemLineMarkerProvid
   protected void collectNavigationMarkers(@NotNull PsiElement element,
                                           @NotNull Collection<? super RelatedItemLineMarkerInfo<?>> result) {
 
-    Optional<PsiClass> classOptional = asReferenceToTypeDecl(element);
+    if (!(element instanceof PsiIdentifier)) {
+      return;
+    }
+
+    Optional<PsiClass> classOptional = asReferenceToTypeDecl(element.getParent());
 
     if (classOptional.isEmpty()) {
       return;