diff --git a/doc/data/messages/m/missing-yield-doc/good.py b/doc/data/messages/m/missing-yield-doc/good.py
index c580aead7c..f9cb9fdfd9 100644
--- a/doc/data/messages/m/missing-yield-doc/good.py
+++ b/doc/data/messages/m/missing-yield-doc/good.py
@@ -1,10 +1,13 @@
-def even_number_under(n: int):
+from typing import Iterator
+
+
+def even_number_under(n: int) -> Iterator[int]:
     """Prints even numbers smaller than n.
     Args:
         n: Upper limit of even numbers.
 
     Yields:
-        int: even numbers
+        even numbers
     """
     for i in range(n):
         if i % 2 == 1: