quick fix to class mismatch