diff --git a/aiida_quantumespresso/workflows/pw/bands.py b/aiida_quantumespresso/workflows/pw/bands.py
index 1a68f46ce..4cf96a7ae 100644
--- a/aiida_quantumespresso/workflows/pw/bands.py
+++ b/aiida_quantumespresso/workflows/pw/bands.py
@@ -194,7 +194,7 @@ def inspect_relax(self):
             return self.exit_codes.ERROR_SUB_PROCESS_FAILED_RELAX
 
         self.ctx.current_structure = workchain.outputs.output_structure
-        self.ctx.current_number_of_bands = workchain.outputs.output_parameters.get_attribute('number_of_bands')
+        self.ctx.current_number_of_bands = workchain.outputs.output_parameters.base.attributes.get('number_of_bands')
 
     def run_seekpath(self):
         """Run the structure through SeeKpath to get the normalized structure and path along high-symmetry k-points .