diff --git a/src/gui_qt/graphs/graphwindow.py b/src/gui_qt/graphs/graphwindow.py
index 0c2ddd1..dea112c 100644
--- a/src/gui_qt/graphs/graphwindow.py
+++ b/src/gui_qt/graphs/graphwindow.py
@@ -111,15 +111,16 @@ class QGraphWindow(QtWidgets.QGraphicsView, Ui_GraphWindow):
         return len(self.active)
 
     def curves(self) -> tuple:
-        for a in self.active:
-            if self.real_button.isChecked():
-                if self.error_plots[a] is not None:
-                    yield self.real_plots[a], self.error_plots[a]
-                else:
-                    yield self.real_plots[a],
+        for set_id in self.sets:
+            if set_id in self.active:
+                if self.real_button.isChecked():
+                    if self.error_plots[set_id] is not None:
+                        yield self.real_plots[set_id], self.error_plots[set_id]
+                    else:
+                        yield self.real_plots[set_id],
 
-            if self.imag_button.isChecked() and self.imag_plots[a] is not None:
-                yield self.imag_plots[a],
+                if self.imag_button.isChecked() and self.imag_plots[set_id] is not None:
+                    yield self.imag_plots[set_id],
 
     @property
     def title(self):