GPII-941: Made a fix in the update function that I had missed when merging