Fixed up update as well