diff --git a/mbtk/mbtk_utils/device_info.c b/mbtk/mbtk_utils/device_info.c
index 864cb36..73bc0d1 100755
--- a/mbtk/mbtk_utils/device_info.c
+++ b/mbtk/mbtk_utils/device_info.c
@@ -1,6 +1,12 @@
 #include <string.h>
 #include "mbtk_type.h"
+
+#ifdef MBTK_DEV_INFO_VERSION_2
+#include "mbtk_device_v2.h"
+#else
 #include "mbtk_device.h"
+#endif
+
 
 static char* band_2_str(mbtk_modem_band_area_enum band_area)
 {