diff --git a/social_norms_trees/atomic_mutations.py b/social_norms_trees/atomic_mutations.py
index 9e62a30..9c74df8 100644
--- a/social_norms_trees/atomic_mutations.py
+++ b/social_norms_trees/atomic_mutations.py
@@ -8,7 +8,7 @@
 import click
 import typer
 
-from behavior_tree_library import Behavior, Composite, Sequence
+from social_norms_trees.behavior_tree_library import Behavior, Composite, Sequence
 
 from pprint import pprint
 
diff --git a/social_norms_trees/interactive_ui.py b/social_norms_trees/interactive_ui.py
index 4aeefc1..93bfb9b 100644
--- a/social_norms_trees/interactive_ui.py
+++ b/social_norms_trees/interactive_ui.py
@@ -7,7 +7,7 @@
 
 from typing import Optional, List, Callable
 
-from behavior_tree_library import Behavior
+from social_norms_trees.behavior_tree_library import Behavior
 
 
 def run_interactive_list(nodes: List, mode: str, new_behavior: Optional[Behavior] = None):
diff --git a/social_norms_trees/ui_wrapper.py b/social_norms_trees/ui_wrapper.py
index f61755b..4e9d0e5 100644
--- a/social_norms_trees/ui_wrapper.py
+++ b/social_norms_trees/ui_wrapper.py
@@ -11,8 +11,8 @@
 
 import typer
 
-from behavior_tree_library import Behavior, Sequence
-from atomic_mutations import remove, insert, move
+from social_norms_trees.behavior_tree_library import Behavior, Sequence
+from social_norms_trees.atomic_mutations import remove, insert, move
 from colorama import Fore, Back, Style
 
 from interactive_ui import run_interactive_list