Fixed error in refactoring file