diff --git a/samples/modules/tflite-micro/magic_wand/src/accelerometer_handler.h b/samples/modules/tflite-micro/magic_wand/src/accelerometer_handler.h
index 630d74676a2..c74382d3cb4 100644
--- a/samples/modules/tflite-micro/magic_wand/src/accelerometer_handler.h
+++ b/samples/modules/tflite-micro/magic_wand/src/accelerometer_handler.h
@@ -19,7 +19,7 @@
 
 #define kChannelNumber 3
 
-#include <tensorflow/lite/c/c_api_internal.h>
+#include <tensorflow/lite/c/c_api_types.h>
 #include <tensorflow/lite/micro/micro_error_reporter.h>
 
 extern int begin_index;
diff --git a/samples/modules/tflite-micro/magic_wand/src/main_functions.cc b/samples/modules/tflite-micro/magic_wand/src/main_functions.cc
index 3bdea24d2b2..12ceaada844 100644
--- a/samples/modules/tflite-micro/magic_wand/src/main_functions.cc
+++ b/samples/modules/tflite-micro/magic_wand/src/main_functions.cc
@@ -25,7 +25,6 @@
 #include <tensorflow/lite/micro/micro_interpreter.h>
 #include <tensorflow/lite/micro/micro_mutable_op_resolver.h>
 #include <tensorflow/lite/schema/schema_generated.h>
-#include <tensorflow/lite/version.h>
 
 /* Globals, used for compatibility with Arduino-style sketches. */
 namespace {