Use tok_ctx.expect() in a few more places.