Take out set_label callback