diff --git a/odl/util/graphics.py b/odl/util/graphics.py
index a83dc7dfff1..98b70c884f3 100644
--- a/odl/util/graphics.py
+++ b/odl/util/graphics.py
@@ -436,14 +436,14 @@ def show_discrete_data(values, grid, title=None, method='',
                 plt.colorbar(mappable=csub, ticks=ticks, format=fmt)
             elif update_in_place:
                 # If it exists and we should update it
-                csub.colorbar.set_clim(minval, maxval)
+                csub.set_clim(minval, maxval)
                 csub.colorbar.set_ticks(ticks)
                 if '%' not in fmt:
                     labels = [fmt] * len(ticks)
                 else:
                     labels = [fmt % t for t in ticks]
                 csub.colorbar.set_ticklabels(labels)
-                csub.colorbar.draw_all()
+                fig.canvas.draw_idle()
 
     # Set title of window
     if title is not None: