diff --git a/Android.bp b/Android.bp
index 28eee94dae..6cd559b647 100644
--- a/Android.bp
+++ b/Android.bp
@@ -175,6 +175,7 @@ android_library {
         "launcher-testing-shared",
         "animationlib",
         "com_android_launcher3_flags_lib",
+        "com_android_wm_shell_flags_lib",
     ],
     sdk_version: "current",
     min_sdk_version: min_launcher3_sdk_version,
diff --git a/tests/Android.bp b/tests/Android.bp
index 34cdbed8d5..0f2a2f8b4d 100644
--- a/tests/Android.bp
+++ b/tests/Android.bp
@@ -97,6 +97,7 @@ android_library {
         "platform-test-rules",
         "testables",
         "com_android_launcher3_flags_lib",
+        "com_android_wm_shell_flags_lib",
     ],
     manifest: "AndroidManifest-common.xml",
     platform_apis: true,