Fix derp from refactoring