save works on first time

app/components/SimpleEditor/SimpleEditor.jsx

@@ -77,7 +77,7 @@ export default class SimpleEditor extends React.Component {
     const exporter = new SimpleExporter(config)
     const convertedSource = exporter.exportDocument(source)
 
-    onSave(convertedSource, callback)
+    onSave(convertedSource)
   }
 
   // NOTE -- Theoretically, we shouldn't lock when the editor is in read only

app/components/SimpleEditor/SimpleEditorExporter.js

@@ -34,11 +34,11 @@ class SimpleExporter extends HTMLExporter {
       throw new Error('Illegal arguments: container is mandatory.')
     }
 
-    var doc = container.getDocument()
+    var doc = container.editorSession.getDocument()
     this.state.doc = doc
     var elements = []
-
-    container.data.nodes.body.nodes.forEach(function (id) {
+    console.log(container)
+    container.editorSession.document.data.nodes.body.nodes.forEach(function (id) {
       var node = doc.get(id)
       var nodeEl = this.convertNode(node)
       elements.push(nodeEl)

app/components/SimpleEditor/SimpleEditorWrapper.jsx

@@ -74,7 +74,7 @@ export class SimpleEditorWrapper extends React.Component {
     fragment.source = source
 
     this.update(fragment)
-    return callback()
+    // return callback()
   }
 
   update (newChapter) {