Missed a refactor