diff --git a/aten.bzl b/aten.bzl
index 9c6325d16abf..6d8cff6d1ae7 100644
--- a/aten.bzl
+++ b/aten.bzl
@@ -62,11 +62,10 @@ def generate_aten_impl(ctx):
     outputs = [ops_dir] + ctx.outputs.outs
 
     install_dir = paths.dirname(ops_dir.path)
-    tool_inputs, tool_inputs_manifest = ctx.resolve_tools(tools = [ctx.attr.generator])
-    ctx.actions.run_shell(
+    ctx.actions.run(
         outputs = outputs,
         inputs = ctx.files.srcs,
-        command = ctx.executable.generator.path + " $@",
+        executable = ctx.executable.generator,
         arguments = [
             "--source-path",
             "aten/src/ATen",
@@ -74,8 +73,6 @@ def generate_aten_impl(ctx):
             "--install_dir",
             install_dir,
         ],
-        tools = tool_inputs,
-        input_manifests = tool_inputs_manifest,
         use_default_shell_env = True,
         mnemonic = "GenerateAten",
     )