diff --git a/src/google/protobuf/editions/golden/editions_transform_proto2.proto b/src/google/protobuf/editions/golden/editions_transform_proto2.proto
index 0b60e80..c8e9fb2 100644
--- a/src/google/protobuf/editions/golden/editions_transform_proto2.proto
+++ b/src/google/protobuf/editions/golden/editions_transform_proto2.proto
@@ -18,7 +18,6 @@
 import "google/protobuf/cpp_features.proto";
 import "google/protobuf/editions/proto/editions_transform_proto3.proto";
 
-option features.enum_type = CLOSED;
 option features.repeated_field_encoding = EXPANDED;
 option features.utf8_validation = NONE;
 option java_multiple_files = true;
@@ -110,6 +109,8 @@
 }
 
 enum TestEnum {
+  option features.enum_type = CLOSED;
+
   FOO = 1; // Non-zero default
 
   BAR = 2;