diff --git a/python/python/ert_gui/simulation/models/ensemble_smoother.py b/python/python/ert_gui/simulation/models/ensemble_smoother.py
index e21c97ef958..69c314ecb47 100644
--- a/python/python/ert_gui/simulation/models/ensemble_smoother.py
+++ b/python/python/ert_gui/simulation/models/ensemble_smoother.py
@@ -1,5 +1,6 @@
 from res.enkf.enums import EnkfInitModeEnum
 from res.enkf.enums import HookRuntime
+from res.enkf.enums import RealizationStateEnum
 from res.enkf import ErtRunContext
 from ert_gui.simulation.models import BaseRunModel, ErtRunError
 
diff --git a/python/python/ert_gui/simulation/models/multiple_data_assimilation.py b/python/python/ert_gui/simulation/models/multiple_data_assimilation.py
index 3591359d856..69fbddf08e2 100644
--- a/python/python/ert_gui/simulation/models/multiple_data_assimilation.py
+++ b/python/python/ert_gui/simulation/models/multiple_data_assimilation.py
@@ -15,6 +15,7 @@
 #  for more details.
 from res.enkf.enums import EnkfInitModeEnum
 from res.enkf.enums import HookRuntime
+from res.enkf.enums import RealizationStateEnum
 from res.enkf import ErtRunContext
 
 from ert_gui.simulation.models import BaseRunModel, ErtRunError