diff --git a/tests/archive/test_archive.py b/tests/archive/test_archive.py
index 74f7a2ae..23ba18c4 100644
--- a/tests/archive/test_archive.py
+++ b/tests/archive/test_archive.py
@@ -201,7 +201,7 @@ def test_writehex(tmpdir, hex_archive):
 
 def test_write_voxel(tmpdir):
     filename = str(tmpdir.join("tmp.cdb"))
-    grid = pv.ImageData(dimensions=(10, 10, 10))
+    grid = pv.UniformGrid(dimensions=(10, 10, 10))
     pymapdl_reader.save_as_archive(filename, grid)
 
     archive = pymapdl_reader.Archive(filename)