Fixing the build

I renamed one of the classes but missed a crucial spec.