another quick fix to class mismatch