diff --git a/build/tasks/dtimage.mk b/build/tasks/dtimage.mk
index 4d6a74d..f102d6c 100644
--- a/build/tasks/dtimage.mk
+++ b/build/tasks/dtimage.mk
@@ -1,7 +1,11 @@
 ifneq ($(filter hikey%, $(TARGET_DEVICE)),)
 ifneq ($(TARGET_NO_DTIMAGE), true)
+# make sure the vendor package is present
 
-MKDTIMG := device/linaro/hikey/installer/hikey960/mkdtimg
+include device/linaro/dragonboard/vendor-package-ver.mk
+ifneq (,$(wildcard $(LINARO_VENDOR_PATH)/hikey960/$(EXPECTED_LINARO_VENDOR_VERSION)/version.mk))
+
+MKDTIMG := $(LINARO_VENDOR_PATH)/hikey960/$(EXPECTED_LINARO_VENDOR_VERSION)/bootloader/mkdtimg
 DTB := $(PRODUCT_OUT)/hi3660-hikey960.dtb
 
 $(PRODUCT_OUT)/dt.img: $(DTB)
@@ -15,3 +19,4 @@
 
 endif
 endif
+endif