diff --git a/tensorflow_gnn/models/gcn/BUILD b/tensorflow_gnn/models/gcn/BUILD
index 83be83f4..8825eebd 100644
--- a/tensorflow_gnn/models/gcn/BUILD
+++ b/tensorflow_gnn/models/gcn/BUILD
@@ -8,6 +8,7 @@ load("@tensorflow_gnn//tensorflow_gnn:tensorflow_gnn.bzl", "tf_py_test")
 licenses(["notice"])
 
 package(
+    default_applicable_licenses = ["//tensorflow_gnn:license"],  # copybara:strip
     default_visibility = [":__subpackages__"],
 )
 
diff --git a/tensorflow_gnn/models/graph_sage/BUILD b/tensorflow_gnn/models/graph_sage/BUILD
index cdb28780..8f17efb1 100644
--- a/tensorflow_gnn/models/graph_sage/BUILD
+++ b/tensorflow_gnn/models/graph_sage/BUILD
@@ -7,10 +7,13 @@ load("@tensorflow_gnn//tensorflow_gnn:tensorflow_gnn.bzl", "tf_py_test")
 
 licenses(["notice"])
 
-package(default_visibility = [
-    "//tensorflow_gnn:__pkg__",
-    "//tensorflow_gnn/graph:__subpackages__",
-])
+package(
+    default_applicable_licenses = ["//tensorflow_gnn:license"],  # copybara:strip
+    default_visibility = [
+        "//tensorflow_gnn:__pkg__",
+        "//tensorflow_gnn/graph:__subpackages__",
+    ],
+)
 
 package_group(name = "users")
 
diff --git a/tensorflow_gnn/models/hgt/BUILD b/tensorflow_gnn/models/hgt/BUILD
index 354803e1..f39ef666 100644
--- a/tensorflow_gnn/models/hgt/BUILD
+++ b/tensorflow_gnn/models/hgt/BUILD
@@ -11,6 +11,7 @@ load("@tensorflow_gnn//tensorflow_gnn:tensorflow_gnn.bzl", "tf_py_test")
 licenses(["notice"])
 
 package(
+    default_applicable_licenses = ["//tensorflow_gnn:license"],  # copybara:strip
     default_visibility = [":__subpackages__"],
 )
 
diff --git a/tensorflow_gnn/models/mt_albis/BUILD b/tensorflow_gnn/models/mt_albis/BUILD
index 4ee7c049..11c121ef 100644
--- a/tensorflow_gnn/models/mt_albis/BUILD
+++ b/tensorflow_gnn/models/mt_albis/BUILD
@@ -11,6 +11,7 @@ load("@tensorflow_gnn//tensorflow_gnn:tensorflow_gnn.bzl", "tf_py_test")
 licenses(["notice"])
 
 package(
+    default_applicable_licenses = ["//tensorflow_gnn:license"],  # copybara:strip
     default_visibility = [":__subpackages__"],
 )
 
diff --git a/tensorflow_gnn/models/multi_head_attention/BUILD b/tensorflow_gnn/models/multi_head_attention/BUILD
index e22db374..f0dd732a 100644
--- a/tensorflow_gnn/models/multi_head_attention/BUILD
+++ b/tensorflow_gnn/models/multi_head_attention/BUILD
@@ -11,6 +11,7 @@ load("@tensorflow_gnn//tensorflow_gnn:tensorflow_gnn.bzl", "tf_py_test")
 licenses(["notice"])
 
 package(
+    default_applicable_licenses = ["//tensorflow_gnn:license"],  # copybara:strip
     default_visibility = [":__subpackages__"],
 )
 
diff --git a/tensorflow_gnn/models/vanilla_mpnn/BUILD b/tensorflow_gnn/models/vanilla_mpnn/BUILD
index fdb8417b..1aaf7b85 100644
--- a/tensorflow_gnn/models/vanilla_mpnn/BUILD
+++ b/tensorflow_gnn/models/vanilla_mpnn/BUILD
@@ -11,6 +11,7 @@ load("@tensorflow_gnn//tensorflow_gnn:tensorflow_gnn.bzl", "tf_py_test")
 licenses(["notice"])
 
 package(
+    default_applicable_licenses = ["//tensorflow_gnn:license"],  # copybara:strip
     default_visibility = [":__subpackages__"],
 )