diff --git a/src/fourc_webviewer/fourc_webserver.py b/src/fourc_webviewer/fourc_webserver.py
index 10dd353..18a996d 100644
--- a/src/fourc_webviewer/fourc_webserver.py
+++ b/src/fourc_webviewer/fourc_webserver.py
@@ -597,7 +597,7 @@ def sync_design_conditions_sections_from_state(self):
         ].sections.items():
             if section_name.startswith("DESIGN "):
                 self._server_vars["fourc_yaml_content"].pop(section_name)
-        self._server_vars["fourc_yaml_content"].add(new_dc_sections)
+        self._server_vars["fourc_yaml_content"].combine_sections(new_dc_sections)
 
     def init_result_description_state_and_server_vars(self):
         """Initialize the state and server variables for the result description
diff --git a/src/fourc_webviewer/run_webserver.py b/src/fourc_webviewer/run_webserver.py
index cea435f..334e504 100644
--- a/src/fourc_webviewer/run_webserver.py
+++ b/src/fourc_webviewer/run_webserver.py
@@ -1,5 +1,6 @@
 """Utility to run the webserver on a defined port."""
 
+from fourc_webviewer.fourc_webserver import FourCWebServer
 from fourc_webviewer_default_files import (
     DEFAULT_INPUT_FILE,
 )