diff --git a/ui-tests/tests/xarray-leaflet.test.ts-snapshots/test0-linux.png b/ui-tests/tests/xarray-leaflet.test.ts-snapshots/test0-linux.png
index 44d0a8a..d37ae08 100644
Binary files a/ui-tests/tests/xarray-leaflet.test.ts-snapshots/test0-linux.png and b/ui-tests/tests/xarray-leaflet.test.ts-snapshots/test0-linux.png differ
diff --git a/xarray_leaflet/transform.py b/xarray_leaflet/transform.py
index 8d1eb0a..6c216fd 100644
--- a/xarray_leaflet/transform.py
+++ b/xarray_leaflet/transform.py
@@ -1,6 +1,6 @@
 import warnings
 import numpy as np
-import xarray as xr
+import xarray.core.rolling
 
 
 def passthrough(array, *args, **kwargs):
@@ -14,7 +14,7 @@ def normalize(array, *args, **kwargs):
     return array
 
 
-def coarsen(agg_func=xr.core.rolling.DataArrayCoarsen.mean):
+def coarsen(agg_func=xarray.core.rolling.DataArrayCoarsen.mean):
     def _(array, *args, **kwargs):
         tile_width = kwargs['tile_width']
         tile_height = kwargs['tile_height']