diff --git a/input/hdi_service/BUILD.gn b/input/hdi_service/BUILD.gn
index 5a3e790644..ba5934e387 100644
--- a/input/hdi_service/BUILD.gn
+++ b/input/hdi_service/BUILD.gn
@@ -47,7 +47,12 @@ ohos_shared_library("libinput_interfaces_service_1.0") {
     ]
   }
 
-  install_images = [ chipset_base_dir, updater_vendor_base_dir ]
+  install_images = [ chipset_base_dir ]
+  if (defined(global_parts_info.updater_ext)) {
+    install_images += [ updater_vendor_base_dir ]
+  } else {
+    install_images += [ "updater" ]
+  }
   subsystem_name = "hdf"
   innerapi_tags = [ "passthrough_indirect" ]
   part_name = "drivers_peripheral_input"