diff --git a/java/java-tests/testData/psi/parser-partial/statements/Yield.txt b/java/java-tests/testData/psi/parser-partial/statements/Yield.txt
index 436004ca8799..3ec79bfd9cf0 100644
--- a/java/java-tests/testData/psi/parser-partial/statements/Yield.txt
+++ b/java/java-tests/testData/psi/parser-partial/statements/Yield.txt
@@ -1,6 +1,6 @@
 PsiJavaFile:Yield.java
   PsiYieldStatement
-    PsiJavaToken:YIELD('yield')
+    PsiKeyword:yield('yield')
     PsiWhiteSpace(' ')
     PsiReferenceExpression:yield
       PsiReferenceParameterList
diff --git a/java/java-tests/testData/psi/parser-partial/statements/YieldIncomplete0.txt b/java/java-tests/testData/psi/parser-partial/statements/YieldIncomplete0.txt
index 723843f6cc63..b4c1466f99e2 100644
--- a/java/java-tests/testData/psi/parser-partial/statements/YieldIncomplete0.txt
+++ b/java/java-tests/testData/psi/parser-partial/statements/YieldIncomplete0.txt
@@ -1,6 +1,6 @@
 PsiJavaFile:YieldIncomplete0.java
   PsiYieldStatement
-    PsiJavaToken:YIELD('yield')
+    PsiKeyword:yield('yield')
     PsiErrorElement:Expression expected
       <empty list>
   PsiWhiteSpace(' ')
\ No newline at end of file
diff --git a/java/java-tests/testData/psi/parser-partial/statements/YieldIncomplete1.txt b/java/java-tests/testData/psi/parser-partial/statements/YieldIncomplete1.txt
index 8885ea82ea33..e8a1d4f056f3 100644
--- a/java/java-tests/testData/psi/parser-partial/statements/YieldIncomplete1.txt
+++ b/java/java-tests/testData/psi/parser-partial/statements/YieldIncomplete1.txt
@@ -1,6 +1,6 @@
 PsiJavaFile:YieldIncomplete1.java
   PsiYieldStatement
-    PsiJavaToken:YIELD('yield')
+    PsiKeyword:yield('yield')
     PsiWhiteSpace(' ')
     PsiReferenceExpression:x
       PsiReferenceParameterList
diff --git a/java/java-tests/testData/psi/parser-partial/statements/YieldNested.txt b/java/java-tests/testData/psi/parser-partial/statements/YieldNested.txt
index d13c940fc4f9..d3a4df83c03c 100644
--- a/java/java-tests/testData/psi/parser-partial/statements/YieldNested.txt
+++ b/java/java-tests/testData/psi/parser-partial/statements/YieldNested.txt
@@ -1,6 +1,6 @@
 PsiJavaFile:YieldNested.java
   PsiYieldStatement
-    PsiJavaToken:YIELD('yield')
+    PsiKeyword:yield('yield')
     PsiWhiteSpace(' ')
     PsiSwitchExpression: 0
       PsiKeyword:switch('switch')
@@ -18,7 +18,7 @@ PsiJavaFile:YieldNested.java
           PsiJavaToken:COLON(':')
         PsiWhiteSpace(' ')
         PsiYieldStatement
-          PsiJavaToken:YIELD('yield')
+          PsiKeyword:yield('yield')
           PsiWhiteSpace(' ')
           PsiLiteralExpression:42
             PsiJavaToken:INTEGER_LITERAL('42')