diff --git a/common/update.c b/common/update.c
index 33bffaa89eca52014e88fdab626912be8f55bbdf..ef61790f4bc83e190ed5dc10ad4161bd71ba55cf 100644
--- a/common/update.c
+++ b/common/update.c
@@ -24,6 +24,7 @@
 #include <malloc.h>
 #include <dfu.h>
 #include <errno.h>
+#include <mtd/cfi_flash.h>
 
 /* env variable holding the location of the update file */
 #define UPDATE_FILE_ENV		"updatefile"