diff --git a/helpers/plot.py b/helpers/plot.py
index 4a742c3beffa46f0a101215e1c2cf259264c6fce..1756edf2a5ea3e278e94f89f87829b8fb039df59 100644
--- a/helpers/plot.py
+++ b/helpers/plot.py
@@ -199,7 +199,7 @@ def plot_reconstruction_2d(data, ground_truth, KL, signal, R, A, name):
         label='reconstruction')
     ax[1, 2].plot(
         amp_mean.domain[0].k_lengths,
-        A.force(ground_truth).to_global_data(),
+        A.force(ground_truth).to_global_data()**2,
         color='b',
         label='ground truth')
     ax[1, 2].legend()