diff --git a/tosa_converter_for_tflite/BUILD b/tosa_converter_for_tflite/BUILD
index f2cfffef978abf6e1da630911b0e432ef7b64866..aafd70c3163506e57f8fc993143408921fb40ca5 100644
--- a/tosa_converter_for_tflite/BUILD
+++ b/tosa_converter_for_tflite/BUILD
@@ -4,7 +4,7 @@
 # SPDX-License-Identifier: Apache-2.0
 #
 
-load("@org_tensorflow//tensorflow:tensorflow.default.bzl", "pybind_extension")
+load("@pybind11_bazel//:build_defs.bzl", "pybind_extension")
 
 cc_library(
     name = "tosa_converter_for_tflite_lib",
@@ -23,7 +23,6 @@ pybind_extension(
     ],
     deps = [
         ":tosa_converter_for_tflite_lib",
-        "@pybind11",
     ],
 )