Fix Bazel build for 05a34dd (#173798)

Commit

13 days ago

Fix Bazel build for 05a34dd (#173798) Co-authored-by: Pranav Kant <prka@google.com>