diff --git a/src/parser/stage3/assignment_expr_parser.rs b/src/parser/stage3/assignment_expr_parser.rs
index 3f7a00a9..f486e118 100644
--- a/src/parser/stage3/assignment_expr_parser.rs
+++ b/src/parser/stage3/assignment_expr_parser.rs
@@ -107,14 +107,15 @@ impl NyashParser {
 #[cfg(test)]
 mod tests {
     use super::*;
-    use crate::tokenizer::Tokenizer;
+    use crate::tokenizer::NyashTokenizer;
 
     #[test]
     fn test_grouped_assignment_simple() {
         std::env::set_var("NYASH_FEATURES", "stage3");
 
         let input = "local y = (x = 42)";
-        let tokens = Tokenizer::tokenize(input).unwrap();
+        let mut tokenizer = NyashTokenizer::new(input);
+        let tokens = tokenizer.tokenize().unwrap();
         let mut parser = NyashParser::new(tokens);
 
         // Skip 'local' and 'y' and '='