diff --git a/tensorflow_probability/python/experimental/linalg/BUILD b/tensorflow_probability/python/experimental/linalg/BUILD
index 86938bbb97..33c960f91e 100644
--- a/tensorflow_probability/python/experimental/linalg/BUILD
+++ b/tensorflow_probability/python/experimental/linalg/BUILD
@@ -123,7 +123,6 @@ multi_substrate_py_library(
 multi_substrate_py_test(
     name = "linear_operator_row_block_test",
     srcs = ["linear_operator_row_block_test.py"],
-    python_version = "PY3",
     deps = [
         ":linear_operator_row_block",
         # numpy dep,
diff --git a/tensorflow_probability/python/sts/BUILD b/tensorflow_probability/python/sts/BUILD
index 2c5c923dfb..454c486cfc 100644
--- a/tensorflow_probability/python/sts/BUILD
+++ b/tensorflow_probability/python/sts/BUILD
@@ -238,7 +238,6 @@ multi_substrate_py_test(
     name = "regularization_test",
     size = "small",
     srcs = ["regularization_test.py"],
-    srcs_version = "PY2AND3",
     deps = [
         ":regularization",
         # numpy dep,