diff --git a/src/bitwarden_to_keepass/BUILD b/src/bitwarden_to_keepass/BUILD
index 56f0f37..ecfc2de 100644
--- a/src/bitwarden_to_keepass/BUILD
+++ b/src/bitwarden_to_keepass/BUILD
@@ -5,6 +5,7 @@ py_library(
     name = "__init__",
     srcs = ["__init__.py"],
     deps = [],
+    imports = [".."],
 )
 
 py_library(
diff --git a/tests/BUILD b/tests/BUILD
index a193995..fed237a 100644
--- a/tests/BUILD
+++ b/tests/BUILD
@@ -26,10 +26,9 @@ py_library(
 pytest_test(
     name = "test_main",
     srcs = ["test_main.py"],
-    # imports = ["src/bitwarden_to_keepass"],
     deps = [
         # To reference something defined in another BUILD file, prefex with
-        # "//"
+        # "//". Note that the path is relative to the top-level WORKSPACE.
         "//src/bitwarden_to_keepass:main",
         # Reference to current BUILD file
         ":__init__",