From cf5b35fbe61801c72ab8756dd8e089c0a037e54c Mon Sep 17 00:00:00 2001
From: "Pavel V. Talanov" <talanov.pavel@gmail.com>
Date: Mon, 31 Mar 2014 14:09:17 +0400
Subject: [PATCH] JetFunctionType#getReturnTypeRef() by stub

---
 .../src/org/jetbrains/jet/lang/psi/JetFunctionType.java        | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetFunctionType.java b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetFunctionType.java
index 919e052c996..a1ca4dec39f 100644
--- a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetFunctionType.java
+++ b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetFunctionType.java
@@ -20,7 +20,6 @@ import com.google.common.collect.Lists;
 import com.intellij.lang.ASTNode;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
-import org.jetbrains.jet.JetNodeTypes;
 import org.jetbrains.jet.lang.psi.stubs.PsiJetPlaceHolderStub;
 import org.jetbrains.jet.lang.psi.stubs.elements.JetStubElementTypes;
 import org.jetbrains.jet.lexer.JetToken;
@@ -87,6 +86,6 @@ public class JetFunctionType extends JetElementImplStub<PsiJetPlaceHolderStub<Je
 
     @Nullable
     public JetTypeReference getReturnTypeRef() {
-        return (JetTypeReference) findChildByType(JetNodeTypes.TYPE_REFERENCE);
+        return getStubOrPsiChild(JetStubElementTypes.TYPE_REFERENCE);
     }
 }