fixed class mismatch