diff --git a/third_party/python/BUILD b/third_party/python/BUILD
index 92400276..46848684 100644
--- a/third_party/python/BUILD
+++ b/third_party/python/BUILD
@@ -86,7 +86,7 @@ python_wheel(
 python_wheel(
     name = "protobuf",
     outs = ["google"],
-    version = "3.12.2",
+    version = "3.18.1",
     deps = [":six"],
 )
 
diff --git a/tools/wheel_resolver/test/BUILD b/tools/wheel_resolver/test/BUILD
index 2c58489a..fe0ed21f 100644
--- a/tools/wheel_resolver/test/BUILD
+++ b/tools/wheel_resolver/test/BUILD
@@ -18,3 +18,20 @@ gentest(
     test_cmd = "PKG=tensorflow; OUTS=$PKG.whl; export OUTS; $DATA --package \"$PKG\"; \
                 if [[ -f \"$PKG\".whl ]]; then exit 0; else exit 1; fi;",
 )
+
+gentest(
+    name = "test_resolve_protobuf",
+    data = ["//tools/wheel_resolver"],
+    no_test_output = True,
+    test_cmd = "true",
+    srcs = [":protobuf"],
+)
+
+python_wheel(
+    name = "protobuf",
+    outs = ["google"],
+    version = "3.18.1",
+    deps = [
+        "//third_party/python:six",
+    ],
+)