diff --git a/app/components/SimpleEditor/elements/track_change/TrackChangeComponent.js b/app/components/SimpleEditor/elements/track_change/TrackChangeComponent.js
index 9fe65a520908f98f3fca973deaf103093c543a3c..4239d21a8d4ccbf2cb16533cec5ec50bad6d31ee 100644
--- a/app/components/SimpleEditor/elements/track_change/TrackChangeComponent.js
+++ b/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
   }