Restore update routine.

This somehow got lost somewhere.