diff --git a/tests/cases/unittests/services/colorization.ts b/tests/cases/unittests/services/colorization.ts
index ec44492db5d..3031cefb56c 100644
--- a/tests/cases/unittests/services/colorization.ts
+++ b/tests/cases/unittests/services/colorization.ts
@@ -256,7 +256,7 @@ describe('Colorization', function () {
                 identifier("Foo"),
                 operator("<"),
                 identifier("Foo"),
-                punctuation(","),
+                operator(","),
                 identifier("Foo"),
                 operator("<"),
                 identifier("number"),
@@ -269,7 +269,7 @@ describe('Colorization', function () {
                 operator("<"),
                 identifier("Foo"),
                 operator(">"),
-                identifier("keyword"),
+                keyword("number"),
                 finalEndOfLineState(ts.EndOfLineState.Start));
         });
     });