Missed a class name from my previous fix.