diff --git a/runtime/queries/wat/highlights.scm b/runtime/queries/wat/highlights.scm
index 007e3bbff..93f03aac0 100644
--- a/runtime/queries/wat/highlights.scm
+++ b/runtime/queries/wat/highlights.scm
@@ -1,4 +1,7 @@
-["module" "func" "param" "result" "type" "memory" "elem" "data" "table" "global"] @keyword
+[
+  "module" "func" "param" "result" "type" "memory" "elem" "data" "table" "global"
+  "if" "then" "else" "block" "loop" "end" "mut"
+] @keyword
 
 ["import" "export"] @keyword.control.import