Missed a broken test

Broken test is fixed