diff --git a/tests/test_write.py b/tests/test_write.py
index 4833369..eced474 100644
--- a/tests/test_write.py
+++ b/tests/test_write.py
@@ -146,7 +146,7 @@ def test_write_zarr_3d(self):
 
             assert tmp.dtype == np.uint8
             assert tmp.sum() == 183750394
-            assert tmp.shape == (1, 1, 1, 2700, 2702)
+            assert tmp.shape == (1, 2700, 2702)
 
     def test_write_zarr_chunk_5d(self):
         """test_write_zarr_5d - Write zarr using TsWriter"""