diff --git a/python/src/com/jetbrains/python/inspections/PyTypeCheckerInspection.java b/python/src/com/jetbrains/python/inspections/PyTypeCheckerInspection.java
index 4908b5db0c18..4fd6dbc3c440 100644
--- a/python/src/com/jetbrains/python/inspections/PyTypeCheckerInspection.java
+++ b/python/src/com/jetbrains/python/inspections/PyTypeCheckerInspection.java
@@ -126,7 +126,7 @@ public class PyTypeCheckerInspection extends PyInspection {
             final PyType expected = myTypeEvalContext.getReturnType(node);
             final String expectedName = PythonDocumentationProvider.getTypeName(expected, myTypeEvalContext);
             if (expected != null && !(expected instanceof PyNoneType)) {
-              registerProblem(annotation != null ? annotation : node.getTypeComment(),
+              registerProblem(annotation != null ? annotation.getValue() : node.getTypeComment(),
                               String.format("Expected to return '%s', got no return", expectedName));
             }
           }
diff --git a/python/testData/inspections/PyTypeCheckerInspection/FunctionReturnTypePy3.py b/python/testData/inspections/PyTypeCheckerInspection/FunctionReturnTypePy3.py
index a008ae0ea8be..57da61172fc1 100644
--- a/python/testData/inspections/PyTypeCheckerInspection/FunctionReturnTypePy3.py
+++ b/python/testData/inspections/PyTypeCheckerInspection/FunctionReturnTypePy3.py
@@ -36,7 +36,7 @@ def h(x) -> int:
 def i() -> Union[int, str]:
     pass
 
-def j(x) <warning descr="Expected to return 'Union[int, str]', got no return">-> Union[int, str]</warning>:
+def j(x) -> <warning descr="Expected to return 'Union[int, str]', got no return">Union[int, str]</warning>:
     x = 42
 
 def k() -> None: