Replace set_elem with put_elem