diff --git a/mbtk/mbtk_utils/mbtk_reboot.c b/mbtk/mbtk_utils/mbtk_reboot.c
index d839ae6..3a85294 100755
--- a/mbtk/mbtk_utils/mbtk_reboot.c
+++ b/mbtk/mbtk_utils/mbtk_reboot.c
@@ -1,6 +1,10 @@
 #include <string.h>
 #include "mbtk_type.h"
+#ifdef MBTK_DEV_INFO_VERSION_2
+#include "mbtk_device_v2.h"
+#else
 #include "mbtk_device.h"
+#endif
 #include "mbtk_utils.h"
 #if 0
 static char* band_2_str(mbtk_modem_band_area_enum band_area)