diff --git a/tests/data/test_case_080/.seismic.segy.yml b/tests/data/test_case_080/.seismic.segy.yml
index 30038b1..1f9a625 100644
--- a/tests/data/test_case_080/.seismic.segy.yml
+++ b/tests/data/test_case_080/.seismic.segy.yml
@@ -18,6 +18,8 @@ fmu:
     description:
     - detailed description
     - optional
+  context:
+    stage: realization
   workflow:
     reference: rms/structural_model
   case:
diff --git a/tests/data/test_case_080/.surface.bin.yml b/tests/data/test_case_080/.surface.bin.yml
index 394a10c..f91bd82 100644
--- a/tests/data/test_case_080/.surface.bin.yml
+++ b/tests/data/test_case_080/.surface.bin.yml
@@ -18,6 +18,8 @@ fmu:
     description:
     - detailed description
     - optional
+  context:
+    stage: realization
   workflow:
     reference: rms/structural_model
   case:
diff --git a/tests/data/test_case_080/.surface_error.bin.yml b/tests/data/test_case_080/.surface_error.bin.yml
index b81c3f3..9739d3a 100644
--- a/tests/data/test_case_080/.surface_error.bin.yml
+++ b/tests/data/test_case_080/.surface_error.bin.yml
@@ -18,6 +18,8 @@ fmu:
     description:
     - detailed description
     - optional
+  context:
+    stage: realization
   workflow:
     reference: rms/structural_model
   case:
diff --git a/tests/data/test_case_080/.surface_restricted.bin.yml b/tests/data/test_case_080/.surface_restricted.bin.yml
index c9ae667..6e81809 100644
--- a/tests/data/test_case_080/.surface_restricted.bin.yml
+++ b/tests/data/test_case_080/.surface_restricted.bin.yml
@@ -18,6 +18,8 @@ fmu:
     description:
     - detailed description
     - optional
+  context:
+    stage: realization
   workflow:
     reference: rms/structural_model
   case:
@@ -160,4 +162,3 @@ masterdata:
     stratigraphic_column:
       identifier: DROGON_2020
       uuid: 12345678-1234-1234-1234-123456789012
-