Add model for config instead of using fromJS

1 files changed, 4 insertions, 3 deletions

diff --git a/packages/gitbook-core/src/reducers/config.js b/packages/gitbook-core/src/reducers/config.js
index a49c602..2ef1ce3 100644
--- a/packages/gitbook-core/src/reducers/config.js
+++ b/packages/gitbook-core/src/reducers/config.js
@@ -1,12 +1,13 @@
-const { fromJS } = require('immutable');
 const ACTION_TYPES = require('../actions/TYPES');
+const Config = require('../models/Config');
 
 module.exports = (state, action) => {
-    state = fromJS(state);
+    state = Config.create(state);
+
     switch (action.type) {
 
     case ACTION_TYPES.PAGE_FETCH_END:
-        return fromJS(action.payload.config);
+        return Config.create(action.payload.config);
 
     default:
         return state;