diff --git a/CMakeLists.txt b/CMakeLists.txt
index cde063a..918d7f1 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -510,6 +510,14 @@ if (ENABLE_CMSISDAP)
 	endif()
 endif(ENABLE_CMSISDAP)
 
+# DFU
+if (ENABLE_DFU)
+	add_definitions(-DENABLE_DFU=1)
+	message("DFU support enabled")
+else()
+	message("DFU support disabled")
+endif()
+
 # dirtyJtag
 if (ENABLE_DIRTYJTAG)
 	add_definitions(-DENABLE_DIRTYJTAG=1)
diff --git a/src/main.cpp b/src/main.cpp
index 225a430..ff99f6e 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -21,7 +21,9 @@
 #include "common.hpp"
 #include "cxxopts.hpp"
 #include "device.hpp"
+#ifdef ENABLE_DFU
 #include "dfu.hpp"
+#endif
 #include "display.hpp"
 #include "efinix.hpp"
 #include "ftdispi.hpp"