diff --git a/src/aiidalab_qe/common/bandpdoswidget.py b/src/aiidalab_qe/common/bandpdoswidget.py
index 51b737318..d1b512a9e 100644
--- a/src/aiidalab_qe/common/bandpdoswidget.py
+++ b/src/aiidalab_qe/common/bandpdoswidget.py
@@ -479,15 +479,15 @@ def handle_spin_polarization(fermi_energy, update_func, layout_type):
         # PDOS plot type
         if self.plot_type == "pdos":
             fermi_energy = self.fermi_energy.get(
-                "fermi_energy"
-            ) or self.fermi_energy.get("fermi_energy_up")
+                "fermi_energy", self.fermi_energy.get("fermi_energy_up")
+            )
             handle_spin_polarization(fermi_energy, fig.update_layout, "layout")
 
         # Combined plot type
         if self.plot_type == "combined":
             fermi_energy = self.fermi_energy.get(
-                "fermi_energy"
-            ) or self.fermi_energy.get("fermi_energy_up")
+                "fermi_energy", self.fermi_energy.get("fermi_energy_up")
+            )
             handle_spin_polarization(fermi_energy, fig.update_xaxes, "xaxes")