Fix onupdate function