Update to work with latest domify component