diff --git a/BoardConfig.mk b/BoardConfig.mk
index 6fd058f..fb0d9f7 100644
--- a/BoardConfig.mk
+++ b/BoardConfig.mk
@@ -12,6 +12,9 @@ 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 08ffb3c..ea17cad 100644
--- a/device.mk
+++ b/device.mk
@@ -32,9 +32,6 @@ 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 \