diff --git a/src/main/gen/io/runescript/plugin/lang/lexer/_RsLexer.java b/src/main/gen/io/runescript/plugin/lang/lexer/_RsLexer.java
index 9842ada..e34f604 100644
--- a/src/main/gen/io/runescript/plugin/lang/lexer/_RsLexer.java
+++ b/src/main/gen/io/runescript/plugin/lang/lexer/_RsLexer.java
@@ -6,7 +6,7 @@
 import com.intellij.lexer.FlexLexer;
 import com.intellij.psi.TokenType;
 import com.intellij.psi.tree.IElementType;
-import it.unimi.dsi.fastutil.ints.IntArrayList;
+import io.runescript.plugin.lang.psi.RsTokenTypes;import it.unimi.dsi.fastutil.ints.IntArrayList;
 import it.unimi.dsi.fastutil.ints.IntStack;
 
 import java.util.List;
@@ -655,7 +655,7 @@ else if (zzAtEOF) {
             switch (zzLexicalState) {
             case BLOCK_COMMENT: {
               popState();
-    return BLOCK_COMMENT;
+    return RsTokenTypes.BLOCK_COMMENT;
             }  // fall though
             case 133: break;
             default:
@@ -886,7 +886,7 @@ else if (zzAtEOF) {
           case 86: break;
           case 37:
             { popState();
-    return BLOCK_COMMENT;
+    return RsTokenTypes.BLOCK_COMMENT;
             }
           // fall through
           case 87: break;
@@ -901,7 +901,7 @@ else if (zzAtEOF) {
           // fall through
           case 89: break;
           case 40:
-            { return BLOCK_COMMENT;
+            { return RsTokenTypes.BLOCK_COMMENT;
             }
           // fall through
           case 90: break;
diff --git a/src/main/grammars/RuneScript.flex b/src/main/grammars/RuneScript.flex
index bbc3ba9..1efbef3 100644
--- a/src/main/grammars/RuneScript.flex
+++ b/src/main/grammars/RuneScript.flex
@@ -3,6 +3,7 @@ package io.runescript.plugin.lang.lexer;
 import com.intellij.lexer.FlexLexer;
 import com.intellij.psi.TokenType;
 import com.intellij.psi.tree.IElementType;
+import io.runescript.plugin.lang.psi.RsTokenTypes;
 import it.unimi.dsi.fastutil.ints.IntArrayList;
 import it.unimi.dsi.fastutil.ints.IntStack;
 
@@ -65,11 +66,11 @@ INCOMPLETE_TAG = "<"(shad|col|str|u|img)"="
 <BLOCK_COMMENT> {
 "*/" {
     popState();
-    return BLOCK_COMMENT;
+    return RsTokenTypes.BLOCK_COMMENT;
 }
 <<EOF>> {
     popState();
-    return BLOCK_COMMENT;
+    return RsTokenTypes.BLOCK_COMMENT;
 }
 [\s\S] { }
 }
@@ -77,7 +78,7 @@ INCOMPLETE_TAG = "<"(shad|col|str|u|img)"="
 <YYINITIAL,STRING_INTERPOLATION> {
 
 // Comments
-"/**/" { return BLOCK_COMMENT; }
+"/**/" { return RsTokenTypes.BLOCK_COMMENT; }
 "/*" { pushState(BLOCK_COMMENT); }
 
 {LINE_COMMENT} { return LINE_COMMENT; }