diff --git a/calculations/PMM.jl b/calculations/PMM.jl
index d1b3cf2..34b4f18 100644
--- a/calculations/PMM.jl
+++ b/calculations/PMM.jl
@@ -75,11 +75,11 @@ end
 all_c = vcat(training_c, extrapolating_c)
 exact_E = [quick_pole_E(V_system(c)) for c in all_c]
 extrapolated_E = ComplexF64[]
-for c in all_c
+for (c, ref) in zip(all_c, exact_E)
     H = H0 + c * H1
     evals, evecs = eigen(H)
     evals = vcat(evals, conj.(evals)) # include complex conjugates
-    push!(extrapolated_E, nearest(evals, exact_E[end]))
+    push!(extrapolated_E, nearest(evals, ref))
 end
 
 # plot results