diff --git a/.gitignore b/.gitignore
index dd15342..e214b18 100644
--- a/.gitignore
+++ b/.gitignore
@@ -38,4 +38,5 @@ MANIFEST.in
 .venv
 /releases/*
 pip-wheel-metadata
-/poetry.toml
\ No newline at end of file
+/poetry.toml
+/venv
\ No newline at end of file
diff --git a/src/xarray_multiscale/multiscale.py b/src/xarray_multiscale/multiscale.py
index 79a3194..85510a9 100644
--- a/src/xarray_multiscale/multiscale.py
+++ b/src/xarray_multiscale/multiscale.py
@@ -297,7 +297,9 @@ def downscale(
         to_downscale.data, reduction, scale_factors, **kwargs
     )
     downscaled_coords = downscale_coords(to_downscale, scale_factors)
-    return DataArray(downscaled_data, downscaled_coords, attrs=array.attrs)
+    return DataArray(
+        downscaled_data, downscaled_coords, attrs=array.attrs, dims=array.dims
+    )
 
 
 def downscale_coords(
diff --git a/tests/test_doc.py b/tests/test_doc.py
new file mode 100644
index 0000000..3aba806
--- /dev/null
+++ b/tests/test_doc.py
@@ -0,0 +1,10 @@
+from xarray_multiscale import multiscale
+from xarray_multiscale.reducers import windowed_mean
+import numpy as np
+import xarray as xr
+
+
+def test_xarray_example():
+    data = xr.DataArray(np.zeros((1024, 1024)), dims=("x", "y"))
+    scaled_data = multiscale(data, windowed_mean, (2, 2))
+    assert len(scaled_data) == 11, "Incorrect number of arrays returned"