diff --git a/wgsl.xml b/wgsl.xml
index 2c8581a..6fad7ef 100644
--- a/wgsl.xml
+++ b/wgsl.xml
@@ -12,30 +12,33 @@
 	<highlighting>
 		<list name="keyword">
 			<item>alias</item>
-			<item>break</item>
-			<item>case</item>
 			<item>const</item>
 			<item>const_assert</item>
-			<item>continue</item>
-			<item>continuing</item>
-			<item>default</item>
 			<item>diagnostic</item>
-			<item>discard</item>
-			<item>else</item>
 			<item>enable</item>
 			<item>false</item>
 			<item>fn</item>
-			<item>for</item>
-			<item>if</item>
 			<item>let</item>
-			<item>loop</item>
 			<item>override</item>
 			<item>requires</item>
-			<item>return</item>
 			<item>struct</item>
-			<item>switch</item>
 			<item>true</item>
 			<item>var</item>
+		</list>
+
+		<list name="control">
+			<item>break</item>
+			<item>case</item>
+			<item>continue</item>
+			<item>continuing</item>
+			<item>default</item>
+			<item>discard</item>
+			<item>else</item>
+			<item>for</item>
+			<item>if</item>
+			<item>loop</item>
+			<item>return</item>
+			<item>switch</item>
 			<item>while</item>
 		</list>
 
@@ -139,6 +142,7 @@
 		<itemDatas>
 			<itemData name="Normal Text" defStyleNum="dsNormal" />
 			<itemData name="Keyword" defStyleNum="dsKeyword" />
+			<itemData name="Control Flow" defStyleNum="dsControlFlow" />
 			<itemData name="BuiltIn" defStyleNum="dsDataType" />
 			<itemData name="Attribute" defStyleNum="dsAttribute" />
 			<itemData name="Comment" defStyleNum="dsComment" />