diff --git a/Android.bp b/Android.bp
index 166ed39..527d246 100644
--- a/Android.bp
+++ b/Android.bp
@@ -198,6 +198,7 @@
         "libcutils",
         "libfs_mgr_binder",
         "libgsi",
+        "libpayload_consumer",
         "libsnapshot",
         "update_metadata-protos",
     ],
@@ -234,11 +235,13 @@
     defaults: [
         "ue_defaults",
         "libupdate_engine_boot_control_exports",
+        "libpayload_consumer_exports",
     ],
     recovery_available: true,
 
     srcs: [
         "boot_control_android.cc",
+        "cleanup_previous_update_action.cc",
         "dynamic_partition_control_android.cc",
         "dynamic_partition_utils.cc",
     ],