Fixed bug in update_ids