diff --git a/lib/ace/mode/_test/tokens_latex.json b/lib/ace/mode/_test/tokens_latex.json
index 0ac37725..77bd5f1a 100644
--- a/lib/ace/mode/_test/tokens_latex.json
+++ b/lib/ace/mode/_test/tokens_latex.json
@@ -6,33 +6,35 @@
   ["rparen","}"]
 ],[
    "start",
-  ["keyword","\\title"],
+  ["storage.type","\\title"],
   ["lparen","{"],
-  ["keyword","\\LaTeX"],
+  ["storage.type","\\LaTeX"],
   ["rparen","}"]
 ],[
    "start",
-  ["keyword","\\date"],
+  ["storage.type","\\date"],
   ["lparen","{"],
   ["rparen","}"]
 ],[
    "start",
-  ["keyword","\\begin"],
-  ["lparen","{"],
-  ["text","document"],
+  ["storage.type","\\begin"],
+  ["lparen","{"]
+],[
+    "block",
+  ["variable.parameter","document"],
   ["rparen","}"]
 ],[
    "start",
   ["text","  "],
-  ["keyword","\\maketitle"]
+  ["storage.type","\\maketitle"]
 ],[
    "start",
   ["text","  "],
-  ["keyword","\\LaTeX"],
+  ["storage.type","\\LaTeX"],
   ["lparen","{"],
   ["rparen","}"],
   ["text"," is a document preparation system for the "],
-  ["keyword","\\TeX"],
+  ["storage.type","\\TeX"],
   ["lparen","{"],
   ["rparen","}"]
 ],[
@@ -50,26 +52,26 @@
 ],[
    "start",
   ["text","  and much more. "],
-  ["keyword","\\LaTeX"],
+  ["storage.type","\\LaTeX"],
   ["lparen","{"],
   ["rparen","}"],
   ["text"," was originally written in 1984 by Leslie"]
 ],[
    "start",
   ["text","  Lamport and has become the dominant method for using "],
-  ["keyword","\\TeX"],
+  ["storage.type","\\TeX"],
   ["text","; few"]
 ],[
    "start",
   ["text","  people write in plain "],
-  ["keyword","\\TeX"],
+  ["storage.type","\\TeX"],
   ["lparen","{"],
   ["rparen","}"],
   ["text"," anymore. The current version  is"]
 ],[
    "start",
   ["text","  "],
-  ["keyword","\\LaTeXe"],
+  ["storage.type","\\LaTeXe"],
   ["text","."]
 ],[
    "start",
@@ -85,43 +87,49 @@
 ],[
    "start",
   ["text","  "],
-  ["keyword","\\begin"],
-  ["lparen","{"],
-  ["text","align"],
+  ["storage.type","\\begin"],
+  ["lparen","{"]
+],[
+    "block",
+  ["variable.parameter","align"],
   ["rparen","}"]
 ],[
    "start",
   ["text","    E &= mc^2                              "],
-  ["keyword","\\\\"]
+  ["constant.character.escape","\\\\"]
 ],[
-   "start",
-  ["text","    m &= "],
-  ["keyword","\\frac"],
+   "equation",
+  ["string","    m &= "],
+  ["constant.character.escape","\\frac"],
   ["lparen","{"],
-  ["text","m_0"],
+  ["string","m_0"],
   ["rparen","}"],
   ["lparen","{"],
-  ["keyword","\\sqrt"],
+  ["constant.character.escape","\\sqrt"],
   ["lparen","{"],
-  ["text","1-"],
-  ["keyword","\\frac"],
+  ["string","1-"],
+  ["constant.character.escape","\\frac"],
   ["lparen","{"],
-  ["text","v^2"],
+  ["string","v^2"],
   ["rparen","}"],
   ["lparen","{"],
-  ["text","c^2"],
+  ["string","c^2"],
   ["rparen","}}}"]
 ],[
    "start",
   ["text","  "],
-  ["keyword","\\end"],
-  ["lparen","{"],
-  ["text","align"],
+  ["storage.type","\\end"],
+  ["lparen","{"]
+],[
+    "block",
+  ["variable.parameter","align"],
   ["rparen","}"]
 ],[
    "start",
-  ["keyword","\\end"],
-  ["lparen","{"],
-  ["text","document"],
+  ["storage.type","\\end"],
+  ["lparen","{"]
+],[
+    "block",
+  ["variable.parameter","document"],
   ["rparen","}"]
 ]]
\ No newline at end of file