diff --git a/runtime/queries/ungrammar/highlights.scm b/runtime/queries/ungrammar/highlights.scm
index 90514a0c5..07d17fcba 100644
--- a/runtime/queries/ungrammar/highlights.scm
+++ b/runtime/queries/ungrammar/highlights.scm
@@ -1,6 +1,12 @@
 (line_comment) @comment
 
-(identifier) @variable
+(identifier) @function
+
+(labeled_rule
+  (identifier) @type)
+
+(node_rule
+  (identifier) @variable.parameter)
 
 (token) @string