Fixed my change in the update operation