diff --git a/src/llm_vm/guided_completion.py b/src/llm_vm/guided_completion.py
index 463f6ba3..5d5b7e81 100644
--- a/src/llm_vm/guided_completion.py
+++ b/src/llm_vm/guided_completion.py
@@ -1,5 +1,5 @@
 import outlines.models as models
-import outlines.text.generate as generate
+import outlines.generate.text as generate
 import torch
 from lark import Lark, Transformer, v_args
 from lark.indenter import PythonIndenter