track toggle view working

app/components/SimpleEditor/elements/track_change/TrackChangeComponent.js

@@ -73,8 +73,8 @@ class TrackChangeComponent extends AnnotationComponent {
     provider.resolve(annotation, action)
   }
 
-  getController () {
-    return this.context.controller
+  getEditor () {
+    return this.context.editor
   }
 
   getProvider () {
@@ -82,9 +82,8 @@ class TrackChangeComponent extends AnnotationComponent {
   }
 
   getViewMode () {
-    const controller = this.getController()
-    // const { trackChangesView } = controller.state
-    const trackChangesView = true
+    const editor = this.getEditor()
+    const { trackChangesView } = editor.state
     return trackChangesView
   }