diff --git a/extensions-builtin/sd_forge_multidiffusion/lib_multidiffusion/tiled_diffusion.py b/extensions-builtin/sd_forge_multidiffusion/lib_multidiffusion/tiled_diffusion.py
index ef8f2a4b..c11bafe3 100644
--- a/extensions-builtin/sd_forge_multidiffusion/lib_multidiffusion/tiled_diffusion.py
+++ b/extensions-builtin/sd_forge_multidiffusion/lib_multidiffusion/tiled_diffusion.py
@@ -578,9 +578,8 @@ class MixtureOfDiffusers(AbstractDiffusion):
                 # controlnet
                 # self.switch_controlnet_tensors(batch_id, N, len(bboxes), is_denoise=True)
                 if 'control' in c_in:
-                    control = c_in['control']
                     self.process_controlnet(x_tile.shape, x_tile.dtype, c_in, cond_or_uncond, bboxes, N, batch_id)
-                    c_tile['control'] = control.get_control(x_tile, t_tile, c_tile, len(cond_or_uncond))
+                    c_tile['control'] = c_in['control_model'].get_control(x_tile, t_tile, c_tile, len(cond_or_uncond))
 
                 # stablesr
                 # self.switch_stablesr_tensors(batch_id)