NOBUG: Added missing set_context call