Hopefully fixed update bug