StreamEx<PsiElement> (@NotNull PsiElement element, @NotNull PyResolveContext context, @NotNull Set<PyReferenceOwner> alreadyVisited) { if (element instanceof PyReferenceExpression && PyAugAssignmentStatementNavigator.getStatementByTarget(element) != null) { return StreamEx.of(resolveReferenceAugAssignmentsAware((PyReferenceOwner)element, context, alreadyVisited)); } return StreamEx.of(element); }