fix is_valid_model_class