Add with_plugin_context util