diff --git a/src/main/java/org/ballerinalang/plugins/idea/psi/FunctionNode.java b/src/main/java/org/ballerinalang/plugins/idea/psi/FunctionNode.java
index f37c9a0f..6a981f6c 100644
--- a/src/main/java/org/ballerinalang/plugins/idea/psi/FunctionNode.java
+++ b/src/main/java/org/ballerinalang/plugins/idea/psi/FunctionNode.java
@@ -96,7 +96,7 @@ public String getPresentableText() {
                     return builder.toString();
                 }
                 // Get the return types.
-                Collection<TypeNameNode> returnTypeNodes = PsiTreeUtil.findChildrenOfType(parameterListNode,
+                Collection<TypeNameNode> returnTypeNodes = PsiTreeUtil.findChildrenOfType(returnTypeListNode,
                         TypeNameNode.class);
                 builder.append(" (");
                 for (TypeNameNode typeNode : returnTypeNodes) {