diff --git a/tests/conftest.py b/tests/conftest.py
index 8ce700e46..4ff7a8224 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -90,7 +90,7 @@ def _create(self):
         lx = len(self.operator.xgrid)
         lpids = len(self.operator.pids)
         for mu2, op in self._operators(
-            mugrid=self.operator.evolgrid, shape=(lpids, lx, lpids, lx)
+            mugrid=self.operator.evolgrid, shape=(lpids, lx)
         ).items():
             self.cache[mu2] = op