Fix update code!