Improve Gitbook.connect to accept a mapActionsToProps

1 files changed, 3 insertions, 3 deletions

diff --git a/packages/gitbook-core/src/lib/renderWithContext.js b/packages/gitbook-core/src/lib/renderWithContext.js
index b9e2cfd..20ec25f 100644
--- a/packages/gitbook-core/src/lib/renderWithContext.js
+++ b/packages/gitbook-core/src/lib/renderWithContext.js
@@ -1,9 +1,9 @@
 const React = require('react');
-const { Provider } = require('react-redux');
 
 const { InjectedComponent } = require('../components/InjectedComponent');
 const PJAXWrapper = require('../components/PJAXWrapper');
 const IntlProvider = require('../components/IntlProvider');
+const ContextProvider = require('../components/IntlProvider');
 
 /**
  * Render the application for a store
@@ -12,13 +12,13 @@ const IntlProvider = require('../components/IntlProvider');
  */
 function renderWithContext(context) {
     return (
-        <Provider store={context.store}>
+        <ContextProvider context={context}>
             <PJAXWrapper>
                 <IntlProvider>
                     <InjectedComponent matching={{ role: 'Body' }} />
                 </IntlProvider>
             </PJAXWrapper>
-        </Provider>
+        </ContextProvider>
     );
 }