Fix wrong refactoring.