diff --git a/Makefile b/Makefile
index aa19cc6..8fe7787 100644
--- a/Makefile
+++ b/Makefile
@@ -1136,7 +1136,7 @@
 	$(call if_changed,mkimage)
 
 ifeq ($(ARCH),arm)
-ifdef CONFIG_DM
+ifdef CONFIG_OF_CONTROL
 UBOOT_BINLOAD := u-boot-dtb.img
 else
 UBOOT_BINLOAD := u-boot.img