missed a couple spots during refactor