diff --git a/flags/Android.bp b/flags/Android.bp
index ba7141d..cb8834a 100644
--- a/flags/Android.bp
+++ b/flags/Android.bp
@@ -21,6 +21,7 @@
 aconfig_declarations {
     name: "wifi_aconfig_flags",
     package: "com.android.wifi.flags",
+    container: "com.android.wifi",
     srcs: ["wifi_flags.aconfig"],
 }
 
diff --git a/flags/wifi_flags.aconfig b/flags/wifi_flags.aconfig
index a35702c..069d812 100644
--- a/flags/wifi_flags.aconfig
+++ b/flags/wifi_flags.aconfig
@@ -1,4 +1,5 @@
 package: "com.android.wifi.flags"
+container: "com.android.wifi"
 
 flag {
     name: "test_flag_function"