fix class error