Fixed bad refactor.