diff --git a/library/flux_models.py b/library/flux_models.py
index d2d7e06c7..35b554681 100644
--- a/library/flux_models.py
+++ b/library/flux_models.py
@@ -1322,7 +1322,7 @@ def forward(
         for block_sample, controlnet_block in zip(block_samples, self.controlnet_blocks):
             block_sample = controlnet_block(block_sample)
             controlnet_block_samples = controlnet_block_samples + (block_sample,)
-        for block_sample, controlnet_block in zip(block_samples, self.controlnet_blocks_for_single):
+        for block_sample, controlnet_block in zip(block_single_samples, self.controlnet_blocks_for_single):
             block_sample = controlnet_block(block_sample)
             controlnet_single_block_samples = controlnet_single_block_samples + (block_sample,)