Fixed update function

lol