fix return statement grammar