diff --git a/src/material/button/BUILD.bazel b/src/material/button/BUILD.bazel
index 036bf223f458..4487e6d3257f 100644
--- a/src/material/button/BUILD.bazel
+++ b/src/material/button/BUILD.bazel
@@ -194,7 +194,11 @@ markdown_to_html(
 
 extract_tokens(
     name = "tokens",
-    srcs = [":theme"],
+    srcs = [
+        ":fab_theme",
+        ":icon_button_theme",
+        ":theme",
+    ],
 )
 
 filegroup(