diff --git a/minimalModel.py b/minimalModel.py
index 32d3506..f3a7af6 100644
--- a/minimalModel.py
+++ b/minimalModel.py
@@ -303,20 +303,19 @@ def surjective_step(data, dim):
 
 
 def build_initial_data(A, dim):
-    g = load("K3_aplKgens")#A.get_cohomology_aplk(A,dim)
-    g = g[2]
+    gens = A.get_cohomology_aplk(A,dim)
     cohomology_gens = A.cohomology_gens(A,dim)
     phi_n_map = {}
     correspondence_n_map = {}
     names = []
     degrees = []
-    for gen_key in g.keys():
+    for gen_key in gens.keys():
         deg = gen_key[0]
         index = gen_key[1]
         new_var = 'x'+str(deg)+'_'+str(index)
         names += [new_var]
         degrees += [deg]
-        phi_n_map[new_var] = g[gen_key]
+        phi_n_map[new_var] = gens[gen_key]
         correspondence_n_map[new_var] = cohomology_gens[gen_key]
     differentials = {}
     gca = GradedCommutativeAlgebra(QQ, names=names, degrees=degrees)