[imp] optimize context handling