Missed a refactored callback in tests