diff --git a/torchquad/integration/gaussian.py b/torchquad/integration/gaussian.py
index 87116184..2bc0a706 100644
--- a/torchquad/integration/gaussian.py
+++ b/torchquad/integration/gaussian.py
@@ -65,7 +65,7 @@ def _weights(self, N, dim, backend, requires_grad=False):
             ).ravel()
         else:
             return anp.prod(
-                anp.meshgrid(*([weights] * dim), indexing="ij", like=backend), axis=0
+                anp.stack(anp.meshgrid(*([weights] * dim), like=backend)), axis=0
             ).ravel()
 
     def _roots(self, N, backend, requires_grad=False):