Fix small refactoring problem.