diff --git a/profiling/graphene.py b/profiling/graphene.py
index 5b653bd343a2c1e64be0f2cfaf9b3aec01ebaba1..ddd3f632bfc6a7b8f3188e6c3c507bb826130d5b 100644
--- a/profiling/graphene.py
+++ b/profiling/graphene.py
@@ -3,6 +3,7 @@ import numpy as np
 from codes.model import Model
 from codes import kwant_examples
 from codes.kwant_helper import utils
+from codes.tb.utils import generate_guess
 from pyinstrument import Profiler
 import timeit
 import memray
@@ -16,7 +17,7 @@ nk = 600
 
 h_int = utils.builder_to_tb(int_builder, params)
 h_0 = utils.builder_to_tb(graphene_builder)
-guess = utils.generate_guess(frozenset(h_int), len(list(h_0.values())[0]))
+guess = generate_guess(frozenset(h_int), len(list(h_0.values())[0]))
 
 model = Model(h_0, h_int, filling)