Fix refactor problem.