diff --git a/callbacks.py b/callbacks.py
index 9902b44..87469ce 100644
--- a/callbacks.py
+++ b/callbacks.py
@@ -29,7 +29,7 @@ def update_figure(value_range, toggle_contours, toggle_black_intervals, selected
         use_black_intervals = 'on' in toggle_black_intervals
         center_on_zero = 'on' in center_colormap
 
-        fig = fct.plot(
+        fig = fct.plot_mesh_with_colorbar(
             vertices, faces, scalars, color_min=min_value, color_max=max_value,
             camera=camera, show_contours=show_contours, colormap=selected_colormap,
             use_black_intervals=use_black_intervals, center_colormap_on_zero=center_on_zero