Fix onUpdate