Fix for multiple divs in edit