adding sebastian mod

aiidateam · Oct 18, 2023 · 0fd373b · 0fd373b
1 parent 7a45f41
commit 0fd373b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/aiida_quantumespresso/workflows/pw/base.py b/src/aiida_quantumespresso/workflows/pw/base.py
@@ -199,8 +199,8 @@ def get_builder_from_protocol(
             parameters = recursive_merge(parameters, parameter_overrides)
 
             # if tot_magnetization in overrides , remove starting_magnetization from parameters
-            if overrides['pw']['parameters']['SYSTEM'].get('tot_magnetization') is not None:
-                parameters['SYSTEM'].pop('starting_magnetization', None)
+            if parameters.get('SYSTEM', {}).get('tot_magnetization') is not None:
+                parameters.setdefault('SYSTEM', {}).pop('starting_magnetization', None)
 
             pseudos_overrides = overrides.get('pw', {}).get('pseudos', {})
             pseudos = recursive_merge(pseudos, pseudos_overrides)