diff --git a/BoardConfig.mk b/BoardConfig.mk
index 94a4e4e..f4c602e 100644
--- a/BoardConfig.mk
+++ b/BoardConfig.mk
@@ -6,6 +6,19 @@
 
 DEVICE_PATH := device/xiaomi/rosemary
 
+# A/B
+AB_OTA_UPDATER := true
+
+AB_OTA_PARTITIONS += \
+    system \
+    product \
+    vendor \
+    boot \
+    vbmeta \
+    vbmeta_system \
+    vbmeta_vendor \
+    dtbo
+
 # APEX
 DEXPREOPT_GENERATE_APEX_IMAGE := true
 
diff --git a/device.mk b/device.mk
index 0ebacdd..7608aef 100644
--- a/device.mk
+++ b/device.mk
@@ -15,6 +15,14 @@ PRODUCT_PACKAGES += \
     android.hardware.boot@1.2-mtkimpl \
     mtk_plpath_utils
 
+PRODUCT_PACKAGES += \
+    update_engine \
+    update_engine_sideload \
+    update_verifier
+
+PRODUCT_PACKAGES_DEBUG += \
+    update_engine_client
+
 # Audio
 PRODUCT_PACKAGES += \
     android.hardware.audio.service \