Fixing a missed class name during refactor.