Fixed an error from the last change with the update method.