diff --git a/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py b/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py
index a4a59726..1b9b690f 100644
--- a/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py
+++ b/extensions-builtin/sd_forge_controlnet/scripts/controlnet.py
@@ -491,6 +491,9 @@ class ControlNetForForgeOfficial(scripts.Script):
     def postprocess_batch_list(self, p, pp, *args, **kwargs):
         for i, unit in enumerate(self.get_enabled_units(args)):
             self.process_unit_after_every_sampling(p, unit, self.current_params[i], pp, *args, **kwargs)
+        return
+
+    def postprocess(self, p, processed, *args):
         self.current_params = {}
         return