PsiElement (@NotNull PyElement element) { final PsiFile file = element.getContainingFile(); if (pyButNotPyiFile(file)) { final PyiFile pythonStubFile = getPythonStubFile((PyFile)file); if (pythonStubFile != null) { return findSimilarElement(element, pythonStubFile); } } return null; }