diff --git a/benchmark.py b/benchmark.py
index 00834b0..84106f8 100644
--- a/benchmark.py
+++ b/benchmark.py
@@ -25,7 +25,7 @@ print(
     round(
         1000000
         * timeit.timeit(
-            "gptc.compile(raw_model, max_ngram_length)",
+            "gptc.Model.compile(raw_model, max_ngram_length)",
             number=compile_iterations,
             globals=globals(),
         )
diff --git a/profiler.py b/profiler.py
index b581951..4891541 100644
--- a/profiler.py
+++ b/profiler.py
@@ -13,4 +13,4 @@ with open("models/raw.json") as f:
 with open("models/benchmark_text.txt") as f:
     text = f.read()
 
-cProfile.run("gptc.compile(raw_model, max_ngram_length)")
+cProfile.run("gptc.Model.compile(raw_model, max_ngram_length)")