Support updating elements on move