diff --git a/BoardConfig.mk b/BoardConfig.mk
index fb0d9f7..6fd058f 100644
--- a/BoardConfig.mk
+++ b/BoardConfig.mk
@@ -12,9 +12,6 @@ DEVICE_PATH := device/xiaomi/umi
 # Display
 TARGET_SCREEN_DENSITY := 440
 
-# Init
-TARGET_INIT_VENDOR_LIB := //$(DEVICE_PATH):init_xiaomi_umi
-
 # Kernel
 TARGET_KERNEL_CONFIG += vendor/xiaomi/umi.config
 
diff --git a/device.mk b/device.mk
index 1961a1d..2ed1791 100644
--- a/device.mk
+++ b/device.mk
@@ -32,6 +32,9 @@ PRODUCT_PACKAGES += \
 PRODUCT_COPY_FILES += \
     $(call find-copy-subdir-files,qdcm_calib_data_*.xml,$(LOCAL_PATH)/qdcm/,$(TARGET_COPY_OUT_VENDOR)/etc/)
 
+# Init
+$(call soong_config_set,libinit,vendor_init_lib,//$(LOCAL_PATH):init_xiaomi_umi)
+
 # Overlays
 DEVICE_PACKAGE_OVERLAYS += \
     $(LOCAL_PATH)/overlay \