diff --git a/python/src/com/jetbrains/python/psi/impl/PyCallExpressionHelper.java b/python/src/com/jetbrains/python/psi/impl/PyCallExpressionHelper.java
index 7d196d6a5386..6a04d80900f1 100644
--- a/python/src/com/jetbrains/python/psi/impl/PyCallExpressionHelper.java
+++ b/python/src/com/jetbrains/python/psi/impl/PyCallExpressionHelper.java
@@ -517,14 +517,7 @@ public class PyCallExpressionHelper {
       }
     }
 
-
-    if (callResultTypes.size() == 1) {
-      return callResultTypes.iterator().next();
-    }
-    if (callResultTypes.size() > 1) {
-      return PyUnionType.union(callResultTypes);
-    }
-    return null;
+    return PyUnionType.union(callResultTypes);
   }
 
   @Nullable