diff --git a/src/Tokenizer/TokensAnalyzer.php b/src/Tokenizer/TokensAnalyzer.php
index 42cde5f85bc..ea9f13dfb6a 100644
--- a/src/Tokenizer/TokensAnalyzer.php
+++ b/src/Tokenizer/TokensAnalyzer.php
@@ -26,6 +26,8 @@
  * @author Gregor Harlan <gharlan@web.de>
  *
  * @internal
+ *
+ * @phpstan-type ClassyElementType 'case'|'const'|'method'|'property'|'trait_import'
  */
 final class TokensAnalyzer
 {
@@ -44,7 +46,7 @@ public function __construct(Tokens $tokens)
     /**
      * Get indices of methods and properties in classy code (classes, interfaces and traits).
      *
-     * @return array<int, array{classIndex: int, token: Token, type: string}>
+     * @return array<int, array{classIndex: int, token: Token, type: ClassyElementType}>
      */
     public function getClassyElements(): array
     {
@@ -724,7 +726,7 @@ public function isSuperGlobal(int $index): bool
      *
      * @param int $classIndex classy index
      *
-     * @return array{int, array<int, array{classIndex: int, token: Token, type: string}>}
+     * @return array{int, array<int, array{classIndex: int, token: Token, type: ClassyElementType}>}
      */
     private function findClassyElements(int $classIndex, int $index): array
     {