Fix @lex_state after parsing def arglist