diff --git a/examples/response_stream/public/mount.tsx b/examples/response_stream/public/mount.tsx
index 7309abf5e0022e..77eebfcd2233dd 100644
--- a/examples/response_stream/public/mount.tsx
+++ b/examples/response_stream/public/mount.tsx
@@ -10,6 +10,7 @@ import * as React from 'react';
 import { render, unmountComponentAtNode } from 'react-dom';
 import { CoreSetup, CoreStart, AppMountParameters } from '@kbn/core/public';
 import { KibanaContextProvider } from '@kbn/kibana-react-plugin/public';
+import { KibanaRenderContextProvider } from '@kbn/react-kibana-context-render';
 import { PLUGIN_NAME } from '../common/constants';
 import { ResponseStreamStartPlugins } from './plugin';
 import { App } from './containers/app';
@@ -41,9 +42,11 @@ export const mount =
     ]);
 
     const reactElement = (
-      <KibanaContextProvider services={deps}>
-        <App />
-      </KibanaContextProvider>
+      <KibanaRenderContextProvider {...core}>
+        <KibanaContextProvider services={deps}>
+          <App />
+        </KibanaContextProvider>
+      </KibanaRenderContextProvider>
     );
     render(reactElement, element);
     return () => unmountComponentAtNode(element);
diff --git a/examples/response_stream/tsconfig.json b/examples/response_stream/tsconfig.json
index 0de5b8c0df5a7f..7a0b99ab1eba2a 100644
--- a/examples/response_stream/tsconfig.json
+++ b/examples/response_stream/tsconfig.json
@@ -22,5 +22,6 @@
     "@kbn/config-schema",
     "@kbn/shared-ux-router",
     "@kbn/ml-response-stream",
+    "@kbn/react-kibana-context-render",
   ]
 }