Bug fixed update() function (2)