diff --git a/src/main/java/dev/openfga/intellijplugin/OpenFGASyntaxHighlighter.java b/src/main/java/dev/openfga/intellijplugin/OpenFGASyntaxHighlighter.java
index f2edf05..f5986d0 100644
--- a/src/main/java/dev/openfga/intellijplugin/OpenFGASyntaxHighlighter.java
+++ b/src/main/java/dev/openfga/intellijplugin/OpenFGASyntaxHighlighter.java
@@ -21,11 +21,11 @@ public class OpenFGASyntaxHighlighter extends SyntaxHighlighterBase {
     public static final TextAttributesKey RELATION_NAME =
             createTextAttributesKey("OPENFGA_RELATION_NAME", DefaultLanguageHighlighterColors.INSTANCE_METHOD);
     public static final TextAttributesKey RELATION_REFERENCE =
-            createTextAttributesKey("OPENFGA_RELATION_REFERENCE", DefaultLanguageHighlighterColors.INSTANCE_METHOD);
+            createTextAttributesKey("OPENFGA_RELATION_REFERENCE", DefaultLanguageHighlighterColors.IDENTIFIER);
     public static final TextAttributesKey TYPE_IDENTIFIER =
             createTextAttributesKey("OPENFGA_TYPE_IDENTIFIER", DefaultLanguageHighlighterColors.CLASS_NAME);
     public static final TextAttributesKey TYPE_REFERENCE =
-            createTextAttributesKey("OPENFGA_TYPE_REFERENCE", DefaultLanguageHighlighterColors.CLASS_REFERENCE);
+            createTextAttributesKey("OPENFGA_TYPE_REFERENCE", DefaultLanguageHighlighterColors.PARAMETER);
     public static final TextAttributesKey CONDITION_IDENTIFIER =
             createTextAttributesKey("OPENFGA_CONDITION_IDENTIFIER", DefaultLanguageHighlighterColors.CLASS_NAME);
     public static final TextAttributesKey CONDITION_REFERENCE =
diff --git a/src/main/resources/OpenFGA_Dark.theme.json b/src/main/resources/OpenFGA_Dark.theme.json
index 9927691..c97583f 100644
--- a/src/main/resources/OpenFGA_Dark.theme.json
+++ b/src/main/resources/OpenFGA_Dark.theme.json
@@ -3,22 +3,7 @@
   "dark": true,
   "author": "",
   "editorScheme": "/OpenFGA_Dark.xml",
-  "colors": {
-    "primaryForeground": "#eeffff",
-    "primaryBackground": "#141517"
-  },
-  "ui": {
-    "*": {
-      "background": "primaryBackground",
-      "separatorColor": "primaryForeground",
-      "foreground": "primaryForeground"
-    },
-    "ToolWindow": {
-      "Header": {
-        "inactiveBackground": "primaryBackground"
-      }
-    }
-  }
+  "ui": {}
 }