Add set_arg_state method to lexer to set after operator state