diff --git a/src/main/kotlin/com/pestphp/pest/PestTestRunLineMarkerProvider.kt b/src/main/kotlin/com/pestphp/pest/PestTestRunLineMarkerProvider.kt
index 5a0a3029..c32f9602 100644
--- a/src/main/kotlin/com/pestphp/pest/PestTestRunLineMarkerProvider.kt
+++ b/src/main/kotlin/com/pestphp/pest/PestTestRunLineMarkerProvider.kt
@@ -2,11 +2,17 @@ package com.pestphp.pest
 
 import com.intellij.execution.lineMarker.RunLineMarkerContributor
 import com.intellij.psi.PsiElement
+import com.jetbrains.php.lang.lexer.PhpTokenTypes
+import com.jetbrains.php.lang.psi.PhpPsiUtil
 import com.jetbrains.php.lang.psi.elements.impl.FunctionReferenceImpl
 
 class PestTestRunLineMarkerProvider : RunLineMarkerContributor() {
     override fun getInfo(leaf: PsiElement): Info? {
-        if (leaf !is FunctionReferenceImpl || !leaf.isPestTestReference()) {
+        if (!PhpPsiUtil.isOfType(leaf, PhpTokenTypes.IDENTIFIER)) {
+            return null
+        }
+
+        if (leaf.parent !is FunctionReferenceImpl || !leaf.parent.isPestTestReference()) {
             return null
         }
 
diff --git a/src/test/kotlin/com/pestphp/pest/PestTestRunLineMarkerProviderTest.kt b/src/test/kotlin/com/pestphp/pest/PestTestRunLineMarkerProviderTest.kt
index aa11bd21..3ff25e8d 100644
--- a/src/test/kotlin/com/pestphp/pest/PestTestRunLineMarkerProviderTest.kt
+++ b/src/test/kotlin/com/pestphp/pest/PestTestRunLineMarkerProviderTest.kt
@@ -11,7 +11,7 @@ class PestTestRunLineMarkerProviderTest : PestLightCodeFixture() {
     fun testMethodCallNamedItAndVariableTestIsNotPestTest() {
         val file = myFixture.configureByFile("MethodCallNamedItAndVariableTest.php")
 
-        val testElement = file.firstChild.lastChild.firstChild
+        val testElement = file.firstChild.lastChild.firstChild.firstChild
 
         assertNull(PestTestRunLineMarkerProvider().getInfo(testElement))
     }
@@ -19,7 +19,7 @@ class PestTestRunLineMarkerProviderTest : PestLightCodeFixture() {
     fun testFunctionCallNamedItWithDescriptionAndClosure() {
         val file = myFixture.configureByFile("PestItFunctionCallWithDescriptionAndClosure.php")
 
-        val testElement = file.firstChild.lastChild.firstChild
+        val testElement = file.firstChild.lastChild.firstChild.firstChild
 
         assertNotNull(PestTestRunLineMarkerProvider().getInfo(testElement))
     }