diff --git a/mace/tools/quantization/BUILD b/mace/tools/quantization/BUILD
index 345bde5d8e236fc1186b547fb0fa60507c09d91c..8ca29e1599f5d79f130761b6f442e3692b55f014 100644
--- a/mace/tools/quantization/BUILD
+++ b/mace/tools/quantization/BUILD
@@ -12,7 +12,7 @@ cc_binary(
     deps = [
         "//external:gflags_nothreads",
         "//mace/codegen:generated_mace_engine_factory",
-        "//mace/codegen:generated_models",
+        "//mace/codegen:libmodels",
         "//mace/libmace",
     ],
 )