Missed a line in the test fix