diff --git a/Android.mk b/Android.mk
index 1b83d75..9e374de 100644
--- a/Android.mk
+++ b/Android.mk
@@ -47,6 +47,11 @@
     install.cpp
 LOCAL_CFLAGS := -Wno-unused-parameter -Werror
 LOCAL_CFLAGS += -DRECOVERY_API_VERSION=$(RECOVERY_API_VERSION)
+
+ifeq ($(AB_OTA_UPDATER),true)
+    LOCAL_CFLAGS += -DAB_OTA_UPDATER=1
+endif
+
 LOCAL_MODULE := librecovery
 LOCAL_STATIC_LIBRARIES := \
     libminui \