fix update_batch