diff --git a/suite2p/gui/gui2p.py b/suite2p/gui/gui2p.py index a1d4e0c6..b9b74856 100644 --- a/suite2p/gui/gui2p.py +++ b/suite2p/gui/gui2p.py @@ -208,7 +208,7 @@ def make_buttons(self): return b0 def roi_text(self, state): - if state == QtCore.Qt.Checked: + if QtCore.Qt.CheckState(state) == QtCore.Qt.Checked: for n in range(len(self.roi_text_labels)): if self.iscell[n] == 1: self.p1.addItem(self.roi_text_labels[n]) @@ -232,7 +232,7 @@ def roi_text(self, state): def zoom_cell(self, state): if self.loaded: - if state == QtCore.Qt.Checked: + if QtCore.Qt.CheckState(state) == QtCore.Qt.Checked: self.zoomtocell = True else: self.zoomtocell = False @@ -554,7 +554,7 @@ def number_chosen(self): self.show() def ROIs_on(self, state): - if state == QtCore.Qt.Checked: + if QtCore.Qt.CheckState(state) == QtCore.Qt.Checked: self.ops_plot["ROIs_on"] = True self.p1.addItem(self.color1) self.p2.addItem(self.color2)