diff --git a/ert_shared/models/ensemble_experiment.py b/ert_shared/models/ensemble_experiment.py
index b8ef5fc4249..1496d1eadd2 100644
--- a/ert_shared/models/ensemble_experiment.py
+++ b/ert_shared/models/ensemble_experiment.py
@@ -46,7 +46,7 @@ def create_context(self, arguments):
         runpath_fmt = model_config.getRunpathFormat( )
         jobname_fmt = model_config.getJobnameFormat( )
         subst_list = self.ert().getDataKW( )
-        itr = arguments["iter_num"]
+        itr = arguments.get("iter_num", 0)
         mask = arguments["active_realizations"]
 
         run_context = ErtRunContext.ensemble_experiment(result_fs,
diff --git a/tests/cli/test_model_factory.py b/tests/cli/test_model_factory.py
index a95ba1cbc40..71b1251528a 100644
--- a/tests/cli/test_model_factory.py
+++ b/tests/cli/test_model_factory.py
@@ -101,6 +101,7 @@ def test_setup_single_test_run(self):
             self.assertTrue(isinstance(model, SingleTestRun))
             self.assertEqual(1, len(argument.keys()))
             self.assertTrue("active_realizations" in argument)
+            model.create_context(argument)
 
     def test_setup_ensemble_experiment(self):
         config_file = self.createTestPath('local/poly_example/poly.ert')
@@ -113,6 +114,7 @@ def test_setup_ensemble_experiment(self):
             self.assertTrue(isinstance(model, EnsembleExperiment))
             self.assertEqual(1, len(argument.keys()))
             self.assertTrue("active_realizations" in argument)
+            model.create_context(argument)
 
     def test_setup_ensemble_smoother(self):
         config_file = self.createTestPath('local/poly_example/poly.ert')
@@ -128,6 +130,7 @@ def test_setup_ensemble_smoother(self):
             self.assertTrue("active_realizations" in argument)
             self.assertTrue("target_case" in argument)
             self.assertTrue("analysis_module" in argument)
+            model.create_context(argument)
 
     def test_setup_multiple_data_assimilation(self):
         config_file = self.createTestPath('local/poly_example/poly.ert')
@@ -135,7 +138,7 @@ def test_setup_multiple_data_assimilation(self):
             ert = work_area.getErt()
             notifier = ErtCliNotifier(ert, config_file)
             ERT.adapt(notifier)
-            args = Namespace(realizations="0-4,7,8", weights="6,4,2", target_case="test_case")
+            args = Namespace(realizations="0-4,7,8", weights="6,4,2", target_case="test_case_%d")
 
             model, argument = model_factory._setup_multiple_data_assimilation(args)
             self.assertTrue(isinstance(model, MultipleDataAssimilation))
@@ -144,6 +147,7 @@ def test_setup_multiple_data_assimilation(self):
             self.assertTrue("target_case" in argument)
             self.assertTrue("analysis_module" in argument)
             self.assertTrue("weights" in argument)
+            model.create_context(argument, 0)
 
     def test_analysis_module_name_iterable(self):