Fix onUpdate function check