1 files changed, 7 insertions, 4 deletions

diff --git a/subprojects/frontend/src/language/problemLanguageSupport.ts b/subprojects/frontend/src/language/problemLanguageSupport.ts
index 2121e05f..3847fdd8 100644
--- a/subprojects/frontend/src/language/problemLanguageSupport.ts
+++ b/subprojects/frontend/src/language/problemLanguageSupport.ts
@@ -27,7 +27,8 @@ const parserWithMetadata = parser.configure({
     styleTags({
       LineComment: t.lineComment,
       BlockComment: t.blockComment,
-      'problem class enum pred fn indiv scope': t.definitionKeyword,
+      'module problem class enum pred fn scope': t.definitionKeyword,
+      'node atom multi': t.definitionKeyword,
       'abstract extends refers contains container opposite': t.modifier,
       'default error contained containment': t.modifier,
       'true false unknown error': t.keyword,
@@ -44,7 +45,7 @@ const parserWithMetadata = parser.configure({
       String: t.string,
       'RelationName/QualifiedName': t.typeName,
       // 'RuleName/QualifiedName': t.typeName,
-      'IndividualNodeName/QualifiedName': t.atom,
+      'AtomNodeName/QualifiedName': t.atom,
       'VariableName/QualifiedName': t.variableName,
       '{ }': t.brace,
       '( )': t.paren,
@@ -54,7 +55,8 @@ const parserWithMetadata = parser.configure({
     }),
     indentNodeProp.add({
       ProblemDeclaration: indentDeclaration,
-      UniqueDeclaration: indentDeclaration,
+      AtomDeclaration: indentDeclaration,
+      NodeDeclaration: indentDeclaration,
       ScopeDeclaration: indentDeclaration,
       PredicateBody: indentPredicateOrRule,
       FunctionBody: indentPredicateOrRule,
@@ -70,7 +72,8 @@ const parserWithMetadata = parser.configure({
       // RuleBody: foldInside,
       Conjunction: foldConjunction,
       // Consequent: foldWholeNode,
-      UniqueDeclaration: foldDeclaration,
+      AtomDeclaration: foldDeclaration,
+      NodeDeclaration: foldDeclaration,
       ScopeDeclaration: foldDeclaration,
       BlockComment: foldBlockComment,
     }),