diff --git a/mediapipe/tasks/cc/text/language_detector/language_detector_test.cc b/mediapipe/tasks/cc/text/language_detector/language_detector_test.cc
index 92dc493e0..2e9a58409 100644
--- a/mediapipe/tasks/cc/text/language_detector/language_detector_test.cc
+++ b/mediapipe/tasks/cc/text/language_detector/language_detector_test.cc
@@ -32,7 +32,7 @@ limitations under the License.
 #include "mediapipe/framework/port/gtest.h"
 #include "mediapipe/framework/port/status_matchers.h"
 #include "mediapipe/tasks/cc/common.h"
-#include "tensorflow/lite/core/shims/cc/shims_test_util.h"
+#include "tensorflow/lite/test_util.h"
 
 namespace mediapipe::tasks::text::language_detector {
 namespace {
@@ -75,7 +75,7 @@ absl::Status MatchesLanguageDetectorResult(
 
 }  // namespace
 
-class LanguageDetectorTest : public tflite_shims::testing::Test {};
+class LanguageDetectorTest : public tflite::testing::Test {};
 
 TEST_F(LanguageDetectorTest, CreateFailsWithMissingModel) {
   auto options = std::make_unique<LanguageDetectorOptions>();
diff --git a/mediapipe/tasks/cc/vision/pose_landmarker/pose_landmarker_test.cc b/mediapipe/tasks/cc/vision/pose_landmarker/pose_landmarker_test.cc
index 062d0746d..87bc97274 100644
--- a/mediapipe/tasks/cc/vision/pose_landmarker/pose_landmarker_test.cc
+++ b/mediapipe/tasks/cc/vision/pose_landmarker/pose_landmarker_test.cc
@@ -38,7 +38,7 @@ limitations under the License.
 #include "mediapipe/tasks/cc/vision/core/image_processing_options.h"
 #include "mediapipe/tasks/cc/vision/pose_landmarker/pose_landmarker_result.h"
 #include "mediapipe/tasks/cc/vision/utils/image_utils.h"
-#include "tensorflow/lite/core/shims/cc/shims_test_util.h"
+#include "tensorflow/lite/test_util.h"
 #include "util/tuple/dump_vars.h"
 
 namespace mediapipe {