diff --git a/fdfd_tools/vectorization.py b/fdfd_tools/vectorization.py
index e7b9645..bdc9d6a 100644
--- a/fdfd_tools/vectorization.py
+++ b/fdfd_tools/vectorization.py
@@ -45,5 +45,5 @@ def unvec(v: vfield_t, shape: numpy.ndarray) -> field_t:
     """
     if numpy.any(numpy.equal(v, None)):
         return None
-    return [vi.reshape(shape, order='C') for vi in numpy.split(v, 3)]
+    return vi.reshape((3, *shape), order='C')