diff --git a/EC.jl b/EC.jl
index 1c320b4..4dbdfc0 100644
--- a/EC.jl
+++ b/EC.jl
@@ -79,7 +79,7 @@ function get_reduced_matrices(EC::affine_EC, training_vecs, gram_schmidt_thresho
 
     if gram_schmidt_threshold > 0; vecs = gram_schmidt!(sampled_vecs, EC.weights, gram_schmidt_threshold; verbose=verbose); end
 
-    EC_basis = hcat(orth_vecs...)
+    EC_basis = hcat(vecs...)
     weights_mat = spdiagm(EC.weights)
     H0_EC = transpose(EC_basis) * weights_mat * EC.H0 * EC_basis
     H1_EC = transpose(EC_basis) * weights_mat * EC.H1 * EC_basis