diff --git a/src/gui_qt/data/container.py b/src/gui_qt/data/container.py
index b4c2d69..24cf09d 100644
--- a/src/gui_qt/data/container.py
+++ b/src/gui_qt/data/container.py
@@ -312,10 +312,8 @@ class ExperimentContainer(QtCore.QObject):
                 err_pen.setColor(QtGui.QColor(*self.plot_real.symbolcolor.rgb()))
                 self.plot_error.setData(pen=err_pen)
 
-        elif mode == 'imag' and self.plot_imag is not None:
+        if mode in ['imag', 'all'] and self.plot_imag is not None:
             self.plot_imag.set_color(color, symbol=symbol, line=line)
-        else:
-            print('Updating color failed for ' + str(self.id))
 
     def setSymbol(self, symbol=None, color=None, size=None, mode='real'):
         if mode in ['real', 'all']: