diff --git a/xarray/tests/test_dask.py b/xarray/tests/test_dask.py
index 6122e987154..52ad14af0f1 100644
--- a/xarray/tests/test_dask.py
+++ b/xarray/tests/test_dask.py
@@ -216,7 +216,7 @@ def test_reduce(self):
         self.assertLazyAndAllClose(u.argmin(dim="x"), actual)
         self.assertLazyAndAllClose((u > 1).any(), (v > 1).any())
         self.assertLazyAndAllClose((u < 1).all("x"), (v < 1).all("x"))
-        with raises_regex(NotImplementedError, "dask"):
+        with raises_regex(NotImplementedError, "along an axis"):
             v.median()
         with raise_if_dask_computes():
             v.reduce(duck_array_ops.mean)