fix(lexer): Return operator when it's the only symbol on the line

1 files changed, 3 insertions, 0 deletions

diff --git a/assembler/src/lexer/mod.rs b/assembler/src/lexer/mod.rs
index b0220fe..c8c8c83 100644
--- a/assembler/src/lexer/mod.rs
+++ b/assembler/src/lexer/mod.rs
@@ -72,6 +72,9 @@ pub fn split(mut line: String) -> Vec<Token> {
                         res.push(Token::Number(num));
                     }
                 }
+                else if res.len() == has_label as usize {
+                    res.push(Token::Operator(buffer.clone()));
+                }
                 else {
                     res.push(Token::Symbol(buffer.clone()));
                 }