diff --git a/tensorflow/lite/experimental/tflite_api_dispatcher/tflite_api_dispatcher.h b/tensorflow/lite/experimental/tflite_api_dispatcher/tflite_api_dispatcher.h
index ecb90b48c50..68ec4378174 100644
--- a/tensorflow/lite/experimental/tflite_api_dispatcher/tflite_api_dispatcher.h
+++ b/tensorflow/lite/experimental/tflite_api_dispatcher/tflite_api_dispatcher.h
@@ -24,8 +24,8 @@ limitations under the License.
 
 // Import the relevant interpreter and model files.
 #if TFLITE_EXPERIMENTAL_RUNTIME
-#include "tensorflow/lite/experimental/tf_runtime/lib/model.h"
-#include "tensorflow/lite/experimental/tf_runtime/public/interpreter.h"
+#include "tensorflow/lite/experimental/tf_runtime/interpreter.h"
+#include "tensorflow/lite/experimental/tf_runtime/model.h"
 #else
 #include "tensorflow/lite/interpreter.h"
 #include "tensorflow/lite/model.h"