Fix example class mismatch