Ensure we preserve context passing for wrapped __call__ functions.