2 files changed, 6 insertions, 1 deletions
diff --git a/packages/gitbook/src/api/decodePage.js b/packages/gitbook/src/api/decodePage.js
index 395c8ff..16e5115 100644
--- a/packages/gitbook/src/api/decodePage.js
+++ b/packages/gitbook/src/api/decodePage.js
@@ -19,7 +19,7 @@ function decodePage(output, page, result) {
     }
 
     // Update page attributes
-    const newAttributes = Immutable.fromJS(page.attributes);
+    const newAttributes = Immutable.fromJS(result.attributes);
     page = page.set('attributes', newAttributes);
 
     // GitBook 3
diff --git a/packages/gitbook/src/api/encodePage.js b/packages/gitbook/src/api/encodePage.js
index 48119f7..7d563cd 100644
--- a/packages/gitbook/src/api/encodePage.js
+++ b/packages/gitbook/src/api/encodePage.js
@@ -23,6 +23,11 @@ function encodePage(output, page) {
     result.path = file.getPath();
     result.rawPath = fs.resolve(result.path);
 
+    result.setAttribute = (key, value) => {
+        result.attributes[key] = value;
+        return result;
+    };
+
     deprecate.field(output, 'page.progress', result, 'progress', function() {
         return encodeProgress(output, page);
     }, '"page.progress" property is deprecated');