Fixed a refactoring oversight