def METHOD_NAME(self, config_and_model_with_extractive_head):
    _, model = config_and_model_with_extractive_head
    head_name = next(iter(EXTRACTIVE_HEAD))
    model.set_task_head(head_name)
    assert model.task_head is model.task_heads[head_name]