diff --git a/b_asic/gui_utils/plot_window.py b/b_asic/gui_utils/plot_window.py
index 023ab152fab4790b5e7430db940e41a49b763e59..7dc5bb114abf27d4758e2762555e5c8ce967ecaa 100644
--- a/b_asic/gui_utils/plot_window.py
+++ b/b_asic/gui_utils/plot_window.py
@@ -22,6 +22,7 @@ from qtpy.QtWidgets import (  # QFrame,; QScrollArea,; QLineEdit,; QSizePolicy,;
     QListWidget,
     QListWidgetItem,
     QPushButton,
+    QSizePolicy,
     QVBoxLayout,
 )
 
@@ -87,11 +88,8 @@ class PlotWindow(QDialog):
 
         self._lines = {}
         for key in sim_res_others | sim_res_delays | sim_res_ins | sim_res_outs:
-            # line = self.plotcanvas.axes.plot(sim_result[key], visible=False, label=key)
             line = self._plot_axes.plot(sim_result[key], label=key)
             self._lines[key] = line[0]
-        # self.plotcanvas.legend = self.plotcanvas.axes.legend()
-        self._legend = self._plot_axes.legend()
 
         self._plot_canvas = FigureCanvas(self._plot_fig)
 
@@ -112,6 +110,7 @@ class PlotWindow(QDialog):
 
         # Add the entire list
         self.checklist = QListWidget()
+        self.checklist.setSizePolicy(QSizePolicy.Preferred, QSizePolicy.Preferred)
         self.checklist.itemChanged.connect(self._item_change)
         listitems = {}
         for key in sim_res_ins | sim_res_outs | sim_res_delays | sim_res_others:
@@ -127,6 +126,7 @@ class PlotWindow(QDialog):
         listlayout.addWidget(self.checklist)
 
         # Add additional checkboxes
+        self._legend = self._plot_axes.legend()
         self.legend_checkbox = QCheckBox("&Legend")
         self.legend_checkbox.stateChanged.connect(self._legend_checkbox_change)
         self.legend_checkbox.setCheckState(Qt.CheckState.Checked)