diff --git a/python/src/com/jetbrains/python/refactoring/move/PyMoveModuleMembersProcessor.java b/python/src/com/jetbrains/python/refactoring/move/PyMoveModuleMembersProcessor.java index f8df8b4a7991..80eab8d1293a 100644 --- a/python/src/com/jetbrains/python/refactoring/move/PyMoveModuleMembersProcessor.java +++ b/python/src/com/jetbrains/python/refactoring/move/PyMoveModuleMembersProcessor.java @@ -245,7 +245,8 @@ public class PyMoveModuleMembersProcessor extends BaseRefactoringProcessor { PyClassRefactoringUtil.updateImportOfElement(importStmt, newElement); } final PsiFile usageFile = usage.getContainingFile(); - if (usageFile == oldElement.getContainingFile() && !PsiTreeUtil.isAncestor(oldElement, usage, false)) { + final PsiElement oldElementBody = PyMoveModuleMembersHelper.expandNamedElementBody(oldElement); + if (usageFile == oldElement.getContainingFile() && !PsiTreeUtil.isAncestor(oldElementBody, usage, false)) { PyClassRefactoringUtil.insertImport(oldElement, newElement); } if (resolvesToLocalStarImport(usage)) {