diff --git a/scripts/bench.fish b/scripts/bench.fish
index e35006e..99633e8 100755
--- a/scripts/bench.fish
+++ b/scripts/bench.fish
@@ -11,7 +11,7 @@ hyperfine \
       --shell none \
       --parameter-list data_path examples/assets/seaCreatures.json,examples/assets/jq_data.json,dielectron.json \
       --warmup 3 \
-      "dust -c '(length (from_json input))' -p {data_path}" \
+      "dust -c 'length(from_json(input))' -p {data_path}" \
       "jq 'length' {data_path}" \
       "node --eval \"require('node:fs').readFile('{data_path}', (err, data)=>{console.log(JSON.parse(data).length)})\"" \
       "nu -c 'open {data_path} | length'"
diff --git a/tree-sitter-dust/highlights.scm b/tree-sitter-dust/highlights.scm
index 879a2b7..59a3259 100644
--- a/tree-sitter-dust/highlights.scm
+++ b/tree-sitter-dust/highlights.scm
@@ -39,7 +39,6 @@
       "else"
       "false"
       "float"
-      "fn"
       "for"
       "if"
       "in"
@@ -50,6 +49,8 @@
       "str"
       "true"
       "while"
+      "->"
+      "=>"
 ] @keyword
 
 (built_in_function) @function.builtin