again fix class error