Fix for onUpdate