diff --git a/helpers/plot.py b/helpers/plot.py
index 44c693e4d529bd0601b6ebc23f9fe9e423bc6f1d..4a742c3beffa46f0a101215e1c2cf259264c6fce 100644
--- a/helpers/plot.py
+++ b/helpers/plot.py
@@ -207,7 +207,7 @@ def plot_reconstruction_2d(data, ground_truth, KL, signal, R, A, name):
     ax[1, 2].set_xscale('log')
     ax[1, 2].set_title('power spectra')
 
-    for c, i, j in enumerate(product(range(2), range(3))):
+    for c, (i, j) in enumerate(product(range(2), range(3))):
         if i != 1 or j != 2:
             ax[i, j].get_xaxis().set_visible(False)
             ax[i, j].get_yaxis().set_visible(False)