diff --git a/lib/gpt/algorithms/optimize/fourier_accelerate.py b/lib/gpt/algorithms/optimize/fourier_accelerate.py
index 8a4d4e90..7b24b42a 100644
--- a/lib/gpt/algorithms/optimize/fourier_accelerate.py
+++ b/lib/gpt/algorithms/optimize/fourier_accelerate.py
@@ -44,7 +44,7 @@ def __init__(self, grid, base, dimensions=None):
             self.weight += c_mu_l
 
         # special consideration for zero
-        self.weight[0, 0, 0, 0] = (2.0 * np.pi) ** 2.0 / np.prod(
+        self.weight[coor[0:1]] = (2.0 * np.pi) ** 2.0 / np.prod(
             [grid.gdimensions[mu] for mu in dimensions]
         ) ** (2.0 / len(dimensions))