diff --git a/test/highlight/shape.d2 b/test/highlight/shape.d2
index fef14ad..59d9638 100644
--- a/test/highlight/shape.d2
+++ b/test/highlight/shape.d2
@@ -1,18 +1,19 @@
 foo.baz.biz
 # <- variable
-#  ^ delimiter
+#  ^ punctuanion.delimiter
 #   ^ variable
-#      ^ delimiter
+#      ^ punctuanion.delimiter
 #       ^ variable
 
 foo: Foo Bar
 # <- variable
-#  ^ delimiter
+#  ^ punctuanion.delimiter
 #    ^ string
 
 foo: Foo Bar {
   # <- variable
   #  ^ label
+  #          ^ punctuation.delimiter
 
   bar.baz
   # <- variable