diff --git a/shepherd/__main__.py b/shepherd/__main__.py
index 5e25588..ed7ca1d 100644
--- a/shepherd/__main__.py
+++ b/shepherd/__main__.py
@@ -1,3 +1,3 @@
 if __name__ == '__main__':
-    from .agent.core import cli
+    from .agent.cli import cli
     cli()