diff --git a/partitionslot/v1_0/BUILD.gn b/partitionslot/v1_0/BUILD.gn
index d4407349efb2661eba0c9ce12fe636df7fddaab3..1992f51e479dcda429e49973a17c5a8b3ea7eab6 100644
--- a/partitionslot/v1_0/BUILD.gn
+++ b/partitionslot/v1_0/BUILD.gn
@@ -11,12 +11,17 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import("//drivers/hdf_core/adapter/uhdf2/hdi.gni")
+import("//build/config/components/hdi/hdi.gni")
 hdi("partitionslot") {
   module_name = "partitionslot_interface_service"
 
   sources = [ "IPartitionSlot.idl" ]
 
+  install_images = [
+      "system",
+      "updater"
+  ]
+
   language = "cpp"
   subsystem_name = "hdf"
   part_name = "drivers_interface_partitionslot"