diff --git a/benchmarks/build_defs.bzl b/benchmarks/build_defs.bzl
index 5f9731d5ff..e3b4125692 100644
--- a/benchmarks/build_defs.bzl
+++ b/benchmarks/build_defs.bzl
@@ -20,7 +20,7 @@ def tmpl_cc_binary(name, gen, args, replacements = [], **kwargs):
     )
 
     if _is_google3:
-        kwargs["malloc"] = "//base:system_malloc"
+        kwargs["malloc"] = "@bazel_tools//tools/cpp:malloc"
         kwargs["features"] = ["-static_linking_mode"]
     native.cc_binary(
         name = name,
diff --git a/editions/BUILD b/editions/BUILD
index 21aba6657d..54e53a373c 100644
--- a/editions/BUILD
+++ b/editions/BUILD
@@ -73,6 +73,7 @@ embed_edition_defaults(
 cc_binary(
     name = "internal_defaults_escape",
     srcs = ["internal_defaults_escape.cc"],
+    malloc = "@bazel_tools//tools/cpp:malloc",
     # This needs to be public for users of embed_edition_defaults.
     visibility = ["//visibility:public"],
     deps = [