diff --git a/tools/releasetools/Android.bp b/tools/releasetools/Android.bp
index 80c233b..3732b78 100644
--- a/tools/releasetools/Android.bp
+++ b/tools/releasetools/Android.bp
@@ -144,7 +144,9 @@
     srcs: [
         "edify_generator.py",
         "ota_from_target_files.py",
+        "target_files_diff.py",
     ],
+    main: "ota_from_target_files.py",
     libs: [
         "releasetools_common",
         "releasetools_verity_utils",