Add fieldset back in