diff --git a/src/Psalm/Checker/StatementsChecker.php b/src/Psalm/Checker/StatementsChecker.php
index 00dac202e..216df05cd 100644
--- a/src/Psalm/Checker/StatementsChecker.php
+++ b/src/Psalm/Checker/StatementsChecker.php
@@ -298,7 +298,7 @@ class StatementsChecker
                         $this->file_name
                     );
 
-                    $function_checker->checkReturnTypes(
+                    $function_checkers[$stmt->name]->checkReturnTypes(
                         false,
                         $return_type,
                         $return_type_location