diff --git a/EC_test.ipynb b/EC_test.ipynb
index 0a14249..c9756ee 100644
--- a/EC_test.ipynb
+++ b/EC_test.ipynb
@@ -16,7 +16,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "vertices = (0, 0.4 - 0.2im, 0.8, 6)\n",
+    "vertices = (0, 0.4 - 0.15im, 0.8, 6)\n",
     "subdivisions = 128\n",
     "p, w = get_mesh(vertices, subdivisions)\n",
     "mesh_E = p.*p ./ (2*0.5)\n",
@@ -53,19 +53,21 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "extrapolate_points = range(0.40, 0.20, 5)\n",
+    "extrapolate_points = range(0.40, 0.25, 5)\n",
     "ref_E = 0.2 - 0.1im\n",
     "\n",
     "exact_E = Vector{ComplexF64}(undef, length(extrapolate_points))\n",
     "extrapolate_E = Vector{ComplexF64}(undef, length(extrapolate_points))\n",
     "\n",
+    "EC_basis_w = EC_basis .* w\n",
+    "N_EC = transpose(EC_basis_w) * EC_basis\n",
+    "\n",
     "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",
-    "    N_EC = transpose(EC_basis) * EC_basis\n",
+    "    H_EC = transpose(EC_basis_w) * H * EC_basis\n",
     "    evals = eigvals(H_EC, N_EC)\n",
     "    i = argmin(abs.(evals .- ref_E))\n",
     "    ref_E = evals[i]\n",