Fix for the fix again

Missed a class name.