diff --git a/tensorflow_model_optimization/python/core/internal/BUILD b/tensorflow_model_optimization/python/core/internal/BUILD
index caaa8ccc..82ed63a2 100644
--- a/tensorflow_model_optimization/python/core/internal/BUILD
+++ b/tensorflow_model_optimization/python/core/internal/BUILD
@@ -1,6 +1,9 @@
 # Placeholder: load py_library
 
-package(default_visibility = ["//visibility:public"])
+package(
+    default_applicable_licenses = ["//third_party/tensorflow_model_optimization:license"],
+    default_visibility = ["//visibility:public"],
+)
 
 licenses(["notice"])