Missed a few changes in update