Fix lexer bugs