Fix DOM after wrapping widget element.