Fix refactoring leftover bug.