add the forgotten update logic