diff --git a/BUILD.bazel b/BUILD.bazel
index 5b719d4..28ed5d7 100644
--- a/BUILD.bazel
+++ b/BUILD.bazel
@@ -156,10 +156,12 @@ cc_test(
     ],
     deps = [
         ":args",
+        ":common",
         ":cross_entropy",
         ":gemma_lib",
         ":ops",
         "@googletest//:gtest_main",
+        "//compression:io",
         "@hwy//:hwy_test_util",
         "@hwy//:thread_pool",
     ],