diff --git a/src/main/python/plotlyst/view/widget/scene/plot.py b/src/main/python/plotlyst/view/widget/scene/plot.py
index 0a6b57810..ced416064 100644
--- a/src/main/python/plotlyst/view/widget/scene/plot.py
+++ b/src/main/python/plotlyst/view/widget/scene/plot.py
@@ -191,19 +191,16 @@ def _changeCharge(self, charge: int):
         self.refresh()
         self.charged.emit()
 
+    @overrides
     def _updateButtons(self):
         if not self.negCharge.isEnabled():
             self.negCharge.setEnabled(True)
-            self.negCharge.setVisible(True)
         if not self.posCharge.isEnabled():
             self.posCharge.setEnabled(True)
-            self.posCharge.setVisible(True)
-        if self.charge == 3:
+        if self.charge() == 3:
             self.posCharge.setDisabled(True)
-            self.posCharge.setHidden(True)
-        if self.charge == -3:
+        if self.charge() == -3:
             self.negCharge.setDisabled(True)
-            self.negCharge.setHidden(True)
 
 
 class ScenePlotSelectorMenu(MenuWidget):