Fix bug in gen_context