2 files changed, 11 insertions, 10 deletions
diff --git a/packages/gitbook-core/src/connect.js b/packages/gitbook-core/src/connect.js
index e30eb22..e18158e 100644
--- a/packages/gitbook-core/src/connect.js
+++ b/packages/gitbook-core/src/connect.js
@@ -6,8 +6,8 @@ const ReactRedux = require('react-redux');
  * @param {Function} mapStateToProps
  * @return {ReactComponent}
  */
-function connect(Component, mapStateToProps) {
-    return ReactRedux.connect(mapStateToProps)(Component);
+function connect(Component, mapStateToProps, mapDispatchToProps) {
+    return ReactRedux.connect(mapStateToProps, mapDispatchToProps)(Component);
 }
 
 module.exports = connect;
diff --git a/packages/gitbook-core/src/reducers/index.js b/packages/gitbook-core/src/reducers/index.js
index 18c9f1e..49ea489 100644
--- a/packages/gitbook-core/src/reducers/index.js
+++ b/packages/gitbook-core/src/reducers/index.js
@@ -1,10 +1,11 @@
-const Redux = require('redux');
+const composeReducer = require('../composeReducer');
+const createReducer = require('../createReducer');
 
-module.exports = Redux.combineReducers({
-    components: require('./components'),
-    navigation: require('./navigation'),
+module.exports = composeReducer(
+    createReducer('components', require('./components')),
+    createReducer('navigation', require('./navigation')),
     // GitBook JSON
-    page:       require('./page'),
-    summary:    require('./summary'),
-    readme:    require('./readme')
-});
+    createReducer('page', require('./page')),
+    createReducer('summary', require('./summary')),
+    createReducer('readme', require('./readme'))
+);