diff --git a/ablescript/src/parser.rs b/ablescript/src/parser.rs
index 4f2bf5a..c829bf9 100644
--- a/ablescript/src/parser.rs
+++ b/ablescript/src/parser.rs
@@ -462,16 +462,10 @@ impl<'source> Parser<'source> {
 
         let mut code: Vec<u8> = vec![];
         loop {
-            match self.checked_next()? {
-                Token::Plus
-                | Token::Minus
-                | Token::Comma
-                | Token::LeftBracket
-                | Token::RightBracket
-                | Token::LessThan
-                | Token::GreaterThan => code.push(self.lexer.slice().as_bytes()[0]),
-                Token::RightCurly => break,
-                _ => (),
+            match self.lexer.next() {
+                Some(Ok(Token::RightCurly)) => break,
+                Some(_) => code.push(self.lexer.slice().as_bytes()[0]),
+                None => return Err(Error::unexpected_eoi(self.lexer.span().start)),
             }
         }