diff --git a/Protobuf.YAML-tmLanguage b/Protobuf.YAML-tmLanguage
index 6d0a2c3..8e465bc 100644
--- a/Protobuf.YAML-tmLanguage
+++ b/Protobuf.YAML-tmLanguage
@@ -46,7 +46,7 @@ repository:
     begin: (enum)(\s+)([A-Za-z][A-Za-z0-9_]*)(\s*)(\{)?
     beginCaptures:
       '1': {name: keyword.source.proto}
-      '3': {name: entity.name.class.proto}
+      '3': {name: entity.name.class.message.proto}
     end: \}
     patterns:
     - include: '#option'
diff --git a/Protobuf.tmLanguage b/Protobuf.tmLanguage
index 44be9bd..966e762 100644
--- a/Protobuf.tmLanguage
+++ b/Protobuf.tmLanguage
@@ -102,7 +102,7 @@
 				<key>3</key>
 				<dict>
 					<key>name</key>
-					<string>entity.name.class.proto</string>
+					<string>entity.name.class.message.proto</string>
 				</dict>
 			</dict>
 			<key>end</key>