diff --git a/projects/datalog-shell/BUILD b/projects/datalog-shell/BUILD
index 66a9af8..5e00b40 100644
--- a/projects/datalog-shell/BUILD
+++ b/projects/datalog-shell/BUILD
@@ -1,6 +1,9 @@
 py_binary(
     name = "datalog-shell",
-    main = "__main__.py",
+    main = "src/python/datalog/shell/__main__.py",
+    imports = [
+        "src/python"
+    ],
     deps = [
         "//projects/datalog",
         py_requirement("prompt_toolkit"),
diff --git a/projects/datalog-shell/__main__.py b/projects/datalog-shell/src/python/datalog/shell/__main__.py
similarity index 100%
rename from projects/datalog-shell/__main__.py
rename to projects/datalog-shell/src/python/datalog/shell/__main__.py