diff --git a/EC_test.ipynb b/EC_test.ipynb
index 157043d..151f8fd 100644
--- a/EC_test.ipynb
+++ b/EC_test.ipynb
@@ -97,6 +97,7 @@
     "end\n",
     "\n",
     "EC_basis = hcat(EC_basis, conj.(EC_basis)) # CA-EC\n",
+    "N_EC = transpose(EC_basis) * EC_basis\n",
     "\n",
     "extrapolate_points = range(0.75, 0.40, 8)\n",
     "\n",
@@ -109,7 +110,6 @@
     "    EC_basis_w = EC_basis .* w\n",
     "    H = get_H_matrix(V_system(c), p, w)\n",
     "    H_EC = transpose(EC_basis) * H * EC_basis\n",
-    "    N_EC = transpose(EC_basis) * EC_basis\n",
     "    evals = eigvals(H_EC, N_EC)\n",
     "    i = argmin(abs.(evals .- exact_E[j]))\n",
     "    extrapolate_E[j] = evals[i]\n",