diff --git a/firmware/dfu.py b/firmware/dfu.py
index 97ead047..d8057505 100644
--- a/firmware/dfu.py
+++ b/firmware/dfu.py
@@ -1,7 +1,8 @@
 import os.path
 import struct
+import sys
 
 with open("_header.bin", "wb") as f:
-    x = struct.pack('<H', os.path.getsize('hackrf_usb_dfu.bin') // 512 + 1)
+    x = struct.pack('<H', os.path.getsize(sys.argv[1] + '_dfu.bin') // 512 + 1)
     y = [0xda, 0xff, x[0], x[1], 0xff, 0xff, 0xff, 0xff]
     f.write(bytearray(y))
diff --git a/firmware/hackrf-common.cmake b/firmware/hackrf-common.cmake
index f8c35b33..a5a9f06a 100644
--- a/firmware/hackrf-common.cmake
+++ b/firmware/hackrf-common.cmake
@@ -194,7 +194,7 @@ macro(DeclareTargets)
 		COMMAND rm -f _tmp.dfu _header.bin
 		COMMAND cp ${PROJECT_NAME}_dfu.bin _tmp.dfu
 		COMMAND ${DFU_COMMAND}
-		COMMAND python ../../dfu.py
+		COMMAND python ../../dfu.py ${PROJECT_NAME}
 		COMMAND cat _header.bin _tmp.dfu >${PROJECT_NAME}.dfu
 		COMMAND rm -f _tmp.dfu _header.bin
 	)