diff --git a/bazel/flags/flags.bzl b/bazel/flags/flags.bzl
index 4823e24919036..95df66854d278 100644
--- a/bazel/flags/flags.bzl
+++ b/bazel/flags/flags.bzl
@@ -35,17 +35,17 @@ _FLAGS = {
         native = lambda ctx: getattr(ctx.fragments.proto, "cc_proto_library_source_suffixes"),
         default = [".pb.cc"],
     ),
-    "proto_toolchain_for_java": struct(
-        native = lambda ctx: getattr(ctx.attr, "_aspect_java_proto_toolchain"),
-        default = "@bazel_tools//tools/proto:java_toolchain",
+    "_proto_toolchain_for_java": struct(
+        native = lambda ctx: "//:java_toolchain",
+        default = "//:java_toolchain",
     ),
-    "proto_toolchain_for_javalite": struct(
-        native = lambda ctx: getattr(ctx.attr, "_aspect_proto_toolchain_for_javalite"),
-        default = "@bazel_tools//tools/proto:javalite_toolchain",
+    "_proto_toolchain_for_javalite": struct(
+        native = lambda ctx: "//:javalite_toolchain",
+        default = "//:javalite_toolchain",
     ),
-    "proto_toolchain_for_cc": struct(
-        native = lambda ctx: getattr(ctx.attr, "_aspect_cc_proto_toolchain"),
-        default = "@bazel_tools//tools/proto:cc_toolchain",
+    "_proto_toolchain_for_cc": struct(
+        native = lambda ctx: "//:cc_toolchain",
+        default = "//:cc_toolchain",
     ),
 }
 
diff --git a/bazel/private/java_lite_proto_library.bzl b/bazel/private/java_lite_proto_library.bzl
index 0f2229030b296..af7ca14b04989 100644
--- a/bazel/private/java_lite_proto_library.bzl
+++ b/bazel/private/java_lite_proto_library.bzl
@@ -72,9 +72,6 @@ _java_lite_proto_aspect = aspect(
     implementation = _aspect_impl,
     attr_aspects = ["deps", "exports"],
     attrs = toolchains.if_legacy_toolchain({
-        _PROTO_TOOLCHAIN_ATTR: attr.label(
-            default = configuration_field(fragment = "proto", name = "proto_toolchain_for_java_lite"),
-        ),
         "_proto_toolchain_for_javalite": attr.label(
             default = Label("//bazel/flags/java:proto_toolchain_for_javalite"),
         ),
@@ -171,9 +168,6 @@ The list of <a href="protocol-buffer.html#proto_library"><code>proto_library</co
 rules to generate Java code for.
 """),
     } | toolchains.if_legacy_toolchain({
-        _PROTO_TOOLCHAIN_ATTR: attr.label(
-            default = configuration_field(fragment = "proto", name = "proto_toolchain_for_java_lite"),
-        ),
         "_proto_toolchain_for_javalite": attr.label(
             default = Label("//bazel/flags/java:proto_toolchain_for_javalite"),
         ),
diff --git a/bazel/private/java_proto_library.bzl b/bazel/private/java_proto_library.bzl
index 7b2f8691e22bc..b0dd56d5229ce 100644
--- a/bazel/private/java_proto_library.bzl
+++ b/bazel/private/java_proto_library.bzl
@@ -73,9 +73,6 @@ java_proto_aspect = aspect(
     implementation = _java_proto_aspect_impl,
     attrs = (
         toolchains.if_legacy_toolchain({
-            "_aspect_java_proto_toolchain": attr.label(
-                default = configuration_field(fragment = "proto", name = "proto_toolchain_for_java"),
-            ),
             "_proto_toolchain_for_java": attr.label(
                 default = "//bazel/flags/java:proto_toolchain_for_java",
             ),
@@ -163,9 +160,6 @@ rules to generate Java code for.
         # buildifier: disable=attr-license (calling attr.license())
         "licenses": attr.license() if hasattr(attr, "license") else attr.string_list(),
     } | toolchains.if_legacy_toolchain({
-        "_aspect_java_proto_toolchain": attr.label(
-            default = configuration_field(fragment = "proto", name = "proto_toolchain_for_java"),
-        ),
         "_proto_toolchain_for_java": attr.label(
             default = "//bazel/flags/java:proto_toolchain_for_java",
         ),
diff --git a/bazel/private/oss/cc_proto_library.bzl b/bazel/private/oss/cc_proto_library.bzl
index d230941628144..3d4444cce2eb3 100644
--- a/bazel/private/oss/cc_proto_library.bzl
+++ b/bazel/private/oss/cc_proto_library.bzl
@@ -125,9 +125,6 @@ cc_proto_aspect = aspect(
                 ),
             } |
             toolchains.if_legacy_toolchain({
-                "_aspect_cc_proto_toolchain": attr.label(
-                    default = configuration_field(fragment = "proto", name = "proto_toolchain_for_cc"),
-                ),
                 "_proto_toolchain_for_cc": attr.label(
                     default = Label("//bazel/flags/cc:proto_toolchain_for_cc"),
                 ),
@@ -196,9 +193,6 @@ The list of <a href="protocol-buffer.html#proto_library"><code>proto_library</co
 rules to generate C++ code for.""",
         ),
     } | toolchains.if_legacy_toolchain({
-        "_aspect_cc_proto_toolchain": attr.label(
-            default = configuration_field(fragment = "proto", name = "proto_toolchain_for_cc"),
-        ),
         "_proto_toolchain_for_cc": attr.label(
             default = "//bazel/flags/cc:proto_toolchain_for_cc",
         ),
diff --git a/bazel/private/proto_lang_toolchain_rule.bzl b/bazel/private/proto_lang_toolchain_rule.bzl
index 8c638065a8e74..f035aab025425 100644
--- a/bazel/private/proto_lang_toolchain_rule.bzl
+++ b/bazel/private/proto_lang_toolchain_rule.bzl
@@ -156,8 +156,7 @@ Deprecated. Alias for <code>denylisted_protos</code>. Will be removed in a futur
         "_proto_compiler": attr.label(
             cfg = "exec",
             executable = True,
-            allow_files = True,
-            default = configuration_field("proto", "proto_compiler"),
+            default = "//bazel/flags:proto_compiler",
         ),
     }),
     provides = [ProtoLangToolchainInfo],