diff --git a/BUILD.bazel b/BUILD.bazel
index 9eb60a4..9f3e952 100644
--- a/BUILD.bazel
+++ b/BUILD.bazel
@@ -261,9 +261,9 @@ cc_library(
         ":threading_context",
         ":tokenizer",
         "//compression:types",
-        "//io",
-        "//io:blob_store",
-        "//io:fields",
+        "//gemma/io",
+        "//gemma/io:blob_store",
+        "//gemma/io:fields",
         "@highway//:hwy",
         "@highway//:profiler",
     ],