Fix mismatched class name