@Override
    public void visitReturnStatement(@NotNull PsiReturnStatement returnStatement) {
      expectedType = PsiTypesUtil.getMethodReturnType(returnStatement);
    }
@Override
    public void visitReturnStatement(@NotNull PsiReturnStatement returnStatement)