Adapt new page.parse api for tests

1 files changed, 4 insertions, 3 deletions

diff --git a/lib/generate/site/index.js b/lib/generate/site/index.js
index 47d09f2..f0a4780 100644
--- a/lib/generate/site/index.js
+++ b/lib/generate/site/index.js
@@ -120,7 +120,7 @@ Generator.prototype.prepareFile  = function(content, _input) {
     .then(function() {
         // Lex, parse includes and get
         // Get HTML generated sections
-        return parse.page(page, {
+        return parse.page(page.content, {
             // Local files path
             dir: path.dirname(_input) || '/',
 
@@ -134,8 +134,9 @@ Generator.prototype.prepareFile  = function(content, _input) {
             ], path.join, fs.readFileSync)
         });
     })
-    .then(function(sections) {
-        page.sections = sections;
+    .then(function(parsed) {
+        page.lexed = parsed.lexed;
+        page.sections = parsed.sections;
 
         // Use plugin hook
         return _callHook("page");