Fixed a refactor.