diff --git a/python/python-psi-impl/src/com/jetbrains/python/documentation/docstrings/DocStringParameterReference.java b/python/python-psi-impl/src/com/jetbrains/python/documentation/docstrings/DocStringParameterReference.java
index ff2fdfac31d5..a25b13650ede 100644
--- a/python/python-psi-impl/src/com/jetbrains/python/documentation/docstrings/DocStringParameterReference.java
+++ b/python/python-psi-impl/src/com/jetbrains/python/documentation/docstrings/DocStringParameterReference.java
@@ -64,12 +64,7 @@ public class DocStringParameterReference extends PsiReferenceBase<PyStringLitera
 
   @Nullable
   private PyTargetExpression resolveGlobalVariable(@NotNull PyFile owner) {
-    for (PyTargetExpression assignment : owner.getTopLevelAttributes()) {
-      if (getCanonicalText().equals(assignment.getName())) {
-        return assignment;
-      }
-    }
-    return null;
+    return owner.findTopLevelAttribute(getCanonicalText());
   }
 
   @Nullable