diff --git a/py/promptlib/promptlib/gen.py b/py/promptlib/promptlib/gen.py
index 5290aa0b..846ed232 100644
--- a/py/promptlib/promptlib/gen.py
+++ b/py/promptlib/promptlib/gen.py
@@ -1,7 +1,7 @@
 from .prompt import PromptNode, TextNode
 
 from typing import List
-from pyaici import ast
+from pyaici import ast_ as ast
 
 class GenNode(PromptNode):
 
diff --git a/py/promptlib/promptlib/prompt.py b/py/promptlib/promptlib/prompt.py
index 95ee827c..d47734e9 100644
--- a/py/promptlib/promptlib/prompt.py
+++ b/py/promptlib/promptlib/prompt.py
@@ -2,7 +2,7 @@
 from typing import List
 from .aici import AICI
 
-from pyaici import ast
+from pyaici import ast_ as ast
 
 class PromptNode:
     """