Fix update of page attributes in hook "page"

1 files changed, 11 insertions, 9 deletions

diff --git a/packages/gitbook-core/src/models/Page.js b/packages/gitbook-core/src/models/Page.js
index 838191e..e3c4a96 100644
--- a/packages/gitbook-core/src/models/Page.js
+++ b/packages/gitbook-core/src/models/Page.js
@@ -1,20 +1,22 @@
-const { Record } = require('immutable');
+const { Record, Map, fromJS } = require('immutable');
 
 const DEFAULTS = {
-    title:    '',
-    content:  '',
-    dir:      'ltr',
-    depth:    1,
-    level:    '',
-    previous: null,
-    next:     null
+    title:      '',
+    content:    '',
+    dir:        'ltr',
+    depth:      1,
+    level:      '',
+    previous:   null,
+    next:       null,
+    attributes: Map()
 };
 
 class Page extends Record(DEFAULTS) {
     static create(state) {
         return state instanceof Page ?
             state : new Page({
-                ...state
+                ...state,
+                attributes: fromJS(state.attributes)
             });
     }
 }