diff --git a/cara/apps/expert.py b/cara/apps/expert.py
index 01f8ef95..779caac2 100644
--- a/cara/apps/expert.py
+++ b/cara/apps/expert.py
@@ -474,7 +474,7 @@ class ExpertApplication:
     @property
     def widget(self):
         self.comparison_view.update_plot()
-        return widgets.VBox(children=(self.tab_widget, self.comparison_view.figure.canvas))
+        return widgets.VBox(children=(self.tab_widget, self.comparison_view.widget))
 
 
 class ComparisonView:
@@ -517,6 +517,10 @@ class ComparisonView:
 
         self.figure.canvas.draw()
 
+    @property
+    def widget(self):
+        return self.figure.canvas
+
 
 def tuple_without_index(t: typing.Tuple, index: int) -> typing.Tuple:
     return t[:index] + t[index + 1:]