diff --git a/src/main/antlr4/astRefactor/astRefactor.g4 b/src/main/antlr4/astRefactor/astRefactor.g4
index db8b6e2b..ef2fe3b4 100644
--- a/src/main/antlr4/astRefactor/astRefactor.g4
+++ b/src/main/antlr4/astRefactor/astRefactor.g4
@@ -4,7 +4,7 @@ grammar astRefactor;
  * Parser Rules
  */
 
-class               : ACCESSMODIFIER WHITESPACE CLASS WHITESPACE NAME '(' MODIFIER WHITESPACE WORD ')''{'NEWLINE classbody'}';
+class               : ACCESSMODIFIER WHITESPACE CLASS WHITESPACE NAME '(' ( MODIFIER| TYPEPLACEHOLDER ) WHITESPACE NAME ')''{'NEWLINE classbody'}';
 classbody           : ( method NEWLINE | variable NEWLINE )+;
 
 variable            : ACCESSMODIFIER WHITESPACE ( MODIFIER WHITESPACE )? NAME WHITESPACE '=' WHITESPACE ( WORD | NUMBER) ';'