Change api for registering components

1 files changed, 5 insertions, 3 deletions

diff --git a/packages/gitbook-core/src/lib/connect.js b/packages/gitbook-core/src/lib/connect.js
index e1b267e..2f2a212 100644
--- a/packages/gitbook-core/src/lib/connect.js
+++ b/packages/gitbook-core/src/lib/connect.js
@@ -2,6 +2,8 @@ const React = require('react');
 const ReactRedux = require('react-redux');
 const { injectIntl } = require('react-intl');
 
+const ContextShape = require('../shapes/Context');
+
 /**
  * Use the GitBook context provided by ContextProvider to map actions to props
  * @param  {ReactComponent} Component
@@ -20,13 +22,13 @@ function connectToActions(Component, mapActionsToProps) {
         },
 
         contextTypes: {
-            gitbookContext: React.PropTypes.object.isRequired
+            gitbook: ContextShape.isRequired
         },
 
         render() {
-            const { gitbookContext } = this.context;
+            const { gitbook } = this.context;
             const { children, ...props } = this.props;
-            const { actions, store } = gitbookContext;
+            const { actions, store } = gitbook;
 
             const actionsProps = mapActionsToProps(actions, store.dispatch);