diff --git a/common/wasm_component_utils.bzl b/common/wasm_component_utils.bzl
index 0bbc1169..05dd1267 100644
--- a/common/wasm_component_utils.bzl
+++ b/common/wasm_component_utils.bzl
@@ -179,7 +179,7 @@ def validate_component_action(ctx, wasm_file, wit_file = None):
     if not (hasattr(ctx.attr, "validate_wit") and ctx.attr.validate_wit):
         return []
 
-    wasm_toolchain = ctx.toolchains.get("@rules_wasm_component//toolchains:wasm_tools_toolchain_type")
+    wasm_toolchain = ctx.toolchains["@rules_wasm_component//toolchains:wasm_tools_toolchain_type"]
     if not wasm_toolchain:
         return []
 
diff --git a/wit/private/wit_bindgen.bzl b/wit/private/wit_bindgen.bzl
index 6d3dfc07..eb7b25f5 100644
--- a/wit/private/wit_bindgen.bzl
+++ b/wit/private/wit_bindgen.bzl
@@ -75,7 +75,7 @@ def _wit_bindgen_impl(ctx):
     # Get wit-bindgen tool from appropriate toolchain
     if ctx.attr.language == "go":
         # Check if TinyGo toolchain is available
-        tinygo_toolchain = ctx.toolchains.get("@rules_wasm_component//toolchains:tinygo_toolchain_type")
+        tinygo_toolchain = ctx.toolchains["@rules_wasm_component//toolchains:tinygo_toolchain_type"]
         if not tinygo_toolchain:
             fail("TinyGo toolchain not available. Go WIT binding generation requires TinyGo toolchain.")
         wit_bindgen = tinygo_toolchain.wit_bindgen_go