diff --git a/Android.bp b/Android.bp
index b740f3d..e615e26 100644
--- a/Android.bp
+++ b/Android.bp
@@ -20,12 +20,12 @@
     "src/trace_processor/metrics/android/android_mem.sql",
     "src/trace_processor/metrics/android/android_mem_lmk.sql",
   ],
-  cmd: "$(location tools/gen_merged_sql_metrics) --cpp_out=$(out) $(in)",
+  cmd: "$(location tools/gen_merged_sql_metrics.py) --cpp_out=$(out) $(in)",
   out: [
     "src/trace_processor/metrics/sql_metrics.h",
   ],
   tool_files: [
-    "tools/gen_merged_sql_metrics",
+    "tools/gen_merged_sql_metrics.py",
   ],
 }