Fix refactoring oversight