diff --git a/tests/test_011_dcs.py b/tests/test_011_dcs.py
index affc3ce2..b7d0793f 100644
--- a/tests/test_011_dcs.py
+++ b/tests/test_011_dcs.py
@@ -8,6 +8,7 @@ def test_dcs_structured():
     line = [(0, 0), (1000, 1000)]
     dcs = nlmod.plot.DatasetCrossSection(ds, line)
     dcs.plot_layers()
+    dcs.label_layers()
     dcs.plot_array(ds["kh"], alpha=0.5)
     dcs.plot_grid()
 
@@ -17,5 +18,6 @@ def test_dcs_vertex():
     line = [(0, 0), (1000, 1000)]
     dcs = nlmod.plot.DatasetCrossSection(ds, line)
     dcs.plot_layers()
+    dcs.label_layers()
     dcs.plot_array(ds["kh"], alpha=0.5)
-    dcs.plot_grid()
+    dcs.plot_grid(vertical=False)