diff --git a/ptml/ptmlTranslateAPI.py b/ptml/ptmlTranslateAPI.py
index fd6212a..47c011d 100644
--- a/ptml/ptmlTranslateAPI.py
+++ b/ptml/ptmlTranslateAPI.py
@@ -16,7 +16,7 @@ class PyTreesAPI():
         
     def prepare(self) -> None:
         # import libraries
-        self.pystr += 'import py_trees\n'
+        self.pystr += 'import py_trees as tree\n'
         
         self.pystr += '\n' * 2