diff --git a/tests/test_write.py b/tests/test_write.py
index e333bd0..55a6af4 100644
--- a/tests/test_write.py
+++ b/tests/test_write.py
@@ -77,8 +77,8 @@ def test_write_zarr_2d(self):
             # Use the temporary directory
             test_file_path = os.path.join(dir, 'out/test.ome.zarr')
 
-            bw = TSWriter(test_file_path)
-            bw.write_image(tmp, tmp.shape, tmp.shape)
+            bw = TSWriter(test_file_path, tmp.shape, tmp.shape, str(tmp.dtype))
+            bw.write_image(tmp)
             bw.close()
 
             br = TSReader(