diff --git a/packages/codemirror-lang-kcl/src/highlight.ts b/packages/codemirror-lang-kcl/src/highlight.ts
index 1b6a8dfb6..e95dfc97d 100644
--- a/packages/codemirror-lang-kcl/src/highlight.ts
+++ b/packages/codemirror-lang-kcl/src/highlight.ts
@@ -29,6 +29,7 @@ export const kclHighlight = styleTags({
   'ObjectProperty/PropertyName': t.definition(t.propertyName),
   'LabeledArgument/ArgumentLabel': t.definition(t.propertyName),
   TagDeclarator: t.tagName,
+  PrimitiveType: t.typeName,
   '( )': t.paren,
   '{ }': t.brace,
   '[ ]': t.bracket,
diff --git a/packages/codemirror-lang-kcl/src/kcl.grammar b/packages/codemirror-lang-kcl/src/kcl.grammar
index 9e9e04781..82667490f 100644
--- a/packages/codemirror-lang-kcl/src/kcl.grammar
+++ b/packages/codemirror-lang-kcl/src/kcl.grammar
@@ -77,7 +77,7 @@ ArgumentList { "(" commaSep<LabeledArgument | expression> ")" }
 type[@isGroup=Type] {
   @specialize[@name=PrimitiveType]<
     identifier,
-    "string" | "number" | "bool" | "sketch" | "sketch_surface" | "solid"
+    "bool" | "number" | "string" | "tag" | "Sketch" | "SketchSurface" | "Solid" | "Plane"
   > |
   ArrayType { type !member "[" "]" } |
   ObjectType { "{" commaSep<ObjectProperty { PropertyName ":" type }> "}" }