Missed a compilation error from refactoring