Fix update code.