diff --git a/tests/Android.bp b/tests/Android.bp
index da447b3..f214f75 100644
--- a/tests/Android.bp
+++ b/tests/Android.bp
@@ -97,6 +97,7 @@
         "platform-test-rules",
         "testables",
         "com_android_launcher3_flags_lib",
+        "com_android_wm_shell_flags_lib",
     ],
     manifest: "AndroidManifest-common.xml",
     platform_apis: true,