diff --git a/datatree/testing.py b/datatree/testing.py
index ebe32cbe..1cbcdf2d 100644
--- a/datatree/testing.py
+++ b/datatree/testing.py
@@ -1,4 +1,4 @@
-from xarray.testing import ensure_warnings
+from xarray.testing.assertions import ensure_warnings
 
 from .datatree import DataTree
 from .formatting import diff_tree_repr