Fix broken refactor.