Fixed the code after extracting RefactoringContext