diff --git a/src/Psalm/Internal/Stubs/CoreGenericClasses.phpstub b/src/Psalm/Internal/Stubs/CoreGenericClasses.phpstub
index fefabd984..0e4425fc2 100644
--- a/src/Psalm/Internal/Stubs/CoreGenericClasses.phpstub
+++ b/src/Psalm/Internal/Stubs/CoreGenericClasses.phpstub
@@ -857,7 +857,7 @@ class ArrayIterator implements SeekableIterator, ArrayAccess, Serializable, Coun
      * Return current array key
      * @link http://php.net/manual/en/arrayiterator.key.php
      *
-     * @return mixed The current array key.
+     * @return TKey The current array key.
      *
      * @since 5.0.0
      */