diff --git a/EC_test.ipynb b/EC_test.ipynb
index 151f8fd..0a14249 100644
--- a/EC_test.ipynb
+++ b/EC_test.ipynb
@@ -97,7 +97,8 @@
     "end\n",
     "\n",
     "EC_basis = hcat(EC_basis, conj.(EC_basis)) # CA-EC\n",
-    "N_EC = transpose(EC_basis) * EC_basis\n",
+    "EC_basis_w = EC_basis .* w\n",
+    "N_EC = transpose(EC_basis_w) * EC_basis\n",
     "\n",
     "extrapolate_points = range(0.75, 0.40, 8)\n",
     "\n",
@@ -107,9 +108,8 @@
     "for (j, c) in enumerate(extrapolate_points)\n",
     "    exact_E[j] = quick_pole_E(V_system(c))\n",
     "\n",
-    "    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",
+    "    H_EC = transpose(EC_basis_w) * H * EC_basis\n",
     "    evals = eigvals(H_EC, N_EC)\n",
     "    i = argmin(abs.(evals .- exact_E[j]))\n",
     "    extrapolate_E[j] = evals[i]\n",