Fix error during fake rendering

1 files changed, 3 insertions, 3 deletions

diff --git a/packages/gitbook-core/src/components/InjectedComponent.js b/packages/gitbook-core/src/components/InjectedComponent.js
index 39b48b2..df2c1c7 100644
--- a/packages/gitbook-core/src/components/InjectedComponent.js
+++ b/packages/gitbook-core/src/components/InjectedComponent.js
@@ -65,11 +65,11 @@ function mapStateToProps(state, props) {
 }
 
 module.exports = {
-    InjectedComponent: ReactRedux.connect(InjectedComponentSet, (state, props) => {
+    InjectedComponent: ReactRedux.connect((state, props) => {
         const result = mapStateToProps(state, props);
         result.components = result.components.slice(0, 1);
         result.withContainer = false;
         return result;
-    }),
-    InjectedComponentSet: ReactRedux.connect(InjectedComponentSet, mapStateToProps)
+    })(InjectedComponentSet),
+    InjectedComponentSet: ReactRedux.connect(mapStateToProps)(InjectedComponentSet)
 };