From 0778644ad607cac80e87220d8b26b3992bbd7598 Mon Sep 17 00:00:00 2001
From: t-reents <timo.reents1@gmail.com>
Date: Mon, 23 Sep 2024 16:21:56 +0200
Subject: [PATCH] Fix support for parent protocol

---
 .../workflows/projwfcbands.py                          | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/src/aiida_wannier90_workflows/workflows/projwfcbands.py b/src/aiida_wannier90_workflows/workflows/projwfcbands.py
index e81966c..6ffa548 100644
--- a/src/aiida_wannier90_workflows/workflows/projwfcbands.py
+++ b/src/aiida_wannier90_workflows/workflows/projwfcbands.py
@@ -118,15 +118,17 @@ def get_builder_from_protocol(  # pylint: disable=arguments-differ
         # Prepare workchain builder
         builder = cls.get_builder()
 
-        projwfc_overrides = None
-        if overrides:
-            projwfc_overrides = overrides.pop("projwfc", None)
+        protocol_inputs = cls.get_protocol_inputs(
+            protocol=protocol, overrides=overrides
+        )
+
+        projwfc_overrides = protocol_inputs.pop("projwfc", None)
 
         pwbands_builder = PwBandsWorkChain.get_builder_from_protocol(
             code=pw_code,
             structure=structure,
             protocol=protocol,
-            overrides=overrides,
+            overrides=protocol_inputs,
             **kwargs,
         )