fix class mismatch