@@ -36,14 +36,15 @@
     # Bookmarks
     - match: '^!!.*|^\[\+\+\].*'
       scope: bookmark
 
     # Graph
-    - match: '^@@@@GRAPH: *(\w+) .*$'
+    - match: '^@@@@GRAPH: *(\w+) *\| *(\w+) .*$'
       scope: graphline
       captures:
         1: string.graphname
+        2: string.graphcode
 
     - match: '^@@@@(?:END_GRAPH .*$| *)'
       scope: graphline
 
     # Keywords are if, else.