diff --git a/tests/test_remap.py b/tests/test_remap.py
index 7d2ac4f..4be904e 100644
--- a/tests/test_remap.py
+++ b/tests/test_remap.py
@@ -95,7 +95,7 @@ def test_combine_da_da():
     new = combine_da_to_da(model, obs, merge=False, interp_time=False)
 
     # Check
-    assert new.dims == {"z": 5, "y": n, "x": n}
+    assert new.sizes == {"z": 5, "y": n, "x": n}
     assert float(new.longitude.min()) == pytest.approx(0.1)
     assert float(new.longitude.max()) == pytest.approx(0.9)
     assert float(new.latitude.min()) == pytest.approx(0.1)
@@ -105,6 +105,6 @@ def test_combine_da_da():
 
     # Use orthogonal selection to get track
     a = new.data.values[:, new.y, new.x]
-    assert a.shape == (model.dims["z"], n), "model levels but obs grid points"
+    assert a.shape == (model.sizes["z"], n), "model levels but obs grid points"
     assert (np.diff(a.mean(axis=0)) >= 0).all(), "obs profile goes S"
     assert np.isclose(np.diff(a.mean(axis=1)), 1, atol=1e-15, rtol=0).all(), "obs profile goes U"