Updated the tests to reflect lexer changes, should pass the CI.