diff --git a/tensorflow/lite/BUILD b/tensorflow/lite/BUILD
index 8e3bec137e4..bc37c645fb7 100644
--- a/tensorflow/lite/BUILD
+++ b/tensorflow/lite/BUILD
@@ -748,7 +748,7 @@ cc_library(
     deps = [
         "//tensorflow/lite:op_resolver",
         "//tensorflow/lite/core/api",
-        "//tensorflow/lite/core/shims:builtin_ops",
+        "//tensorflow/lite/kernels:builtin_ops",
     ],
 )
 
diff --git a/tensorflow/lite/create_op_resolver_with_builtin_ops.cc b/tensorflow/lite/create_op_resolver_with_builtin_ops.cc
index 0eda0b743ae..fb8b6eaeb2b 100644
--- a/tensorflow/lite/create_op_resolver_with_builtin_ops.cc
+++ b/tensorflow/lite/create_op_resolver_with_builtin_ops.cc
@@ -15,8 +15,8 @@ limitations under the License.
 
 #include <memory>
 
-#include "tensorflow/lite/core/shims/cc/kernels/register.h"
 #include "tensorflow/lite/create_op_resolver.h"
+#include "tensorflow/lite/kernels/register.h"
 
 namespace tflite {