Make callback on update call optional