diff --git a/tools/aconfig/aconfig/src/codegen/java.rs b/tools/aconfig/aconfig/src/codegen/java.rs
index 4ec6615db9..d3f074a555 100644
--- a/tools/aconfig/aconfig/src/codegen/java.rs
+++ b/tools/aconfig/aconfig/src/codegen/java.rs
@@ -482,7 +482,7 @@ mod tests {
                 .unwrap();
         let expect_flags_content = EXPECTED_FLAG_COMMON_CONTENT.to_string()
             + r#"
-            private static final FeatureFlags FEATURE_FLAGS = new FeatureFlagsImpl();
+            private static FeatureFlags FEATURE_FLAGS = new FeatureFlagsImpl();
         }"#;
 
         let expect_featureflagsimpl_content = r#"
@@ -670,7 +670,7 @@ mod tests {
             public static boolean enabledRoExported() {
                 return FEATURE_FLAGS.enabledRoExported();
             }
-            private static final FeatureFlags FEATURE_FLAGS = new FeatureFlagsImpl();
+            private static FeatureFlags FEATURE_FLAGS = new FeatureFlagsImpl();
         }
         "#;
 
@@ -1092,7 +1092,7 @@ mod tests {
             public static boolean enabledRw() {
                 return FEATURE_FLAGS.enabledRw();
             }
-            private static final FeatureFlags FEATURE_FLAGS = new FeatureFlagsImpl();
+            private static FeatureFlags FEATURE_FLAGS = new FeatureFlagsImpl();
         }"#;
 
         let expect_customfeatureflags_content = r#"
diff --git a/tools/aconfig/aconfig/templates/Flags.java.template b/tools/aconfig/aconfig/templates/Flags.java.template
index 2d012f572d..e2f70b95fa 100644
--- a/tools/aconfig/aconfig/templates/Flags.java.template
+++ b/tools/aconfig/aconfig/templates/Flags.java.template
@@ -33,11 +33,8 @@ public final class Flags \{
     public static void unsetFeatureFlags() \{
         Flags.FEATURE_FLAGS = null;
     }
-
-    private static FeatureFlags FEATURE_FLAGS;
-{{ -else }}
-
-    private static final FeatureFlags FEATURE_FLAGS = new FeatureFlagsImpl();
 {{ -endif }}
 
+    private static FeatureFlags FEATURE_FLAGS{{ -if not is_test_mode }} = new FeatureFlagsImpl(){{ -endif- }};
+
 }