Pass a predefined function to update

Instead of wrapping the update function, use a predefined
method.