diff --git a/jax_tpu_embedding/sparsecore/examples/models/shakespeare/BUILD b/jax_tpu_embedding/sparsecore/examples/models/shakespeare/BUILD
index 9a729cd..3b2c3fa 100644
--- a/jax_tpu_embedding/sparsecore/examples/models/shakespeare/BUILD
+++ b/jax_tpu_embedding/sparsecore/examples/models/shakespeare/BUILD
@@ -24,7 +24,6 @@ pytype_strict_library(
     srcs = [
         "dataset.py",
     ],
-    srcs_version = "PY3",
     deps = [
         pypi_requirement("absl/logging"),
         pypi_requirement("numpy"),
@@ -36,7 +35,6 @@ pytype_strict_library(
     srcs = [
         "model.py",
     ],
-    srcs_version = "PY3",
     deps = [
         "//jax_tpu_embedding/sparsecore/lib/nn:embedding_spec",
         pypi_requirement("flax:core"),
@@ -50,7 +48,6 @@ pytype_strict_library(
     srcs = [
         "flax_model.py",
     ],
-    srcs_version = "PY3",
     deps = [
         "//jax_tpu_embedding/sparsecore/lib/flax:embed",
         "//jax_tpu_embedding/sparsecore/lib/nn:embedding",
diff --git a/jax_tpu_embedding/sparsecore/lib/core/BUILD b/jax_tpu_embedding/sparsecore/lib/core/BUILD
index 16285a6..e966c92 100644
--- a/jax_tpu_embedding/sparsecore/lib/core/BUILD
+++ b/jax_tpu_embedding/sparsecore/lib/core/BUILD
@@ -93,7 +93,6 @@ pytype_strict_library(
     srcs = [
         "input_preprocessing.py",
     ],
-    srcs_version = "PY3",
     deps = [
         ":constants",
         pypi_requirement("absl:app"),
@@ -109,7 +108,6 @@ pytype_strict_library(
     srcs = [
         "constants.py",
     ],
-    srcs_version = "PY3",
 )
 
 pytype_strict_contrib_test(