Fixed an error with the refactor.