diff --git a/y2018/control_loops/python/BUILD b/y2018/control_loops/python/BUILD
index ba34d86..7cea41b 100644
--- a/y2018/control_loops/python/BUILD
+++ b/y2018/control_loops/python/BUILD
@@ -169,7 +169,7 @@
         "graph_edit.py",
         "graph_generate.py",
     ],
-    default_python_version = "PY3",
+    python_version = "PY3",
     legacy_create_init = False,
     restricted_to = ["//tools:k8"],
     srcs_version = "PY3",
@@ -187,7 +187,7 @@
         "graph_codegen.py",
         "graph_generate.py",
     ],
-    default_python_version = "PY2",
+    python_version = "PY2",
     legacy_create_init = False,
     srcs_version = "PY2",
     deps = [