diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD
index b522c5044c8..cad8ccaaad6 100644
--- a/tensorflow/python/BUILD
+++ b/tensorflow/python/BUILD
@@ -3247,6 +3247,7 @@ py_library(
         "layers/base.py",
         "layers/utils.py",
     ],
+    srcs_version = "PY2AND3",
     deps = [
         ":array_ops",
         ":control_flow_ops",