diff --git a/compareECLFiles.cmake b/compareECLFiles.cmake
index 78402a4f1..60c950f36 100644
--- a/compareECLFiles.cmake
+++ b/compareECLFiles.cmake
@@ -165,12 +165,12 @@ add_test_compareECLFiles(CASENAME spe9group
                          ABS_TOL ${abs_tol}
                          REL_TOL ${rel_tol})
 
-# TODO: move to flow
 add_test_compareECLFiles(CASENAME msw_2d_h
                          FILENAME 2D_H__
-                         SIMULATOR flow_multisegment
+                         SIMULATOR flow
                          ABS_TOL ${abs_tol}
-                         REL_TOL ${rel_tol})
+                         REL_TOL ${rel_tol}
+                         TEST_ARGS use_multisegment_well=true)
 
 # TODO: add flow
 add_test_compareECLFiles(CASENAME polymer_simple2D
diff --git a/tests/update_reference_data.sh b/tests/update_reference_data.sh
index 4c9155bd8..fab803cf3 100755
--- a/tests/update_reference_data.sh
+++ b/tests/update_reference_data.sh
@@ -73,8 +73,8 @@ for test_name in ${tests}; do
   if grep -q "msw_2d_h" <<< $test_name
   then
     copyToReferenceDir \
-      $configuration/build-opm-simulators/tests/results/flow_multisegment+msw_2d_h/ \
-      $OPM_DATA_ROOT/msw_2d_h/opm-simulation-reference/flow_multisegment \
+      $configuration/build-opm-simulators/tests/results/flow+msw_2d_h/ \
+      $OPM_DATA_ROOT/msw_2d_h/opm-simulation-reference/flow \
       2D_H__ \
       EGRID INIT SMSPEC UNRST UNSMRY
   fi