Fix for JS onUpdate