Fix model_name inheritance