diff --git a/queries/highlights.scm b/queries/highlights.scm
index 628eea6..6cd4076 100644
--- a/queries/highlights.scm
+++ b/queries/highlights.scm
@@ -1,4 +1,9 @@
 ; Keywords {{{
+[
+ "function"
+ "primitive"
+] @keyword.function
+
 [
   "array"
   (break_expression)
diff --git a/test/highlight/keywords.tig b/test/highlight/keywords.tig
index 122d91a..5ead351 100644
--- a/test/highlight/keywords.tig
+++ b/test/highlight/keywords.tig
@@ -5,9 +5,9 @@ let
   /* <- keyword */
 
   function f() : int = a
-  /* <- keyword */
+  /* <- keyword.function */
   primitive g()
-  /* <- keyword */
+  /* <- keyword.function */
 
   import "lib.tih"
   /* <- keyword */