diff --git a/custom_ext/BUILD b/custom_ext/BUILD
index 19c8650..16e2c54 100644
--- a/custom_ext/BUILD
+++ b/custom_ext/BUILD
@@ -1,5 +1,6 @@
 load("@pybind11_bazel//:build_defs.bzl", "pybind_extension")
-load("@rules_python//python:defs.bzl", "py_library", "py_test")
+load("@rules_python//python:defs.bzl", "py_test")
+load("@rules_python//python:py_library.bzl", "py_library")
 
 package(default_visibility = ["//visibility:private"])