Fix folding after moving parent
Made elements re-arrange after being dragged around