diff --git a/helios/diffusers_version/pipeline_helios_diffusers.py b/helios/diffusers_version/pipeline_helios_diffusers.py
index 37624f5..72e0a08 100644
--- a/helios/diffusers_version/pipeline_helios_diffusers.py
+++ b/helios/diffusers_version/pipeline_helios_diffusers.py
@@ -724,7 +724,7 @@ class HeliosPipeline(DiffusionPipeline, HeliosLoraLoaderMixin):
                     )[0]
 
                 if self.do_classifier_free_guidance:
-                    with self.transformer.cache_context("cond_uncond"):
+                    with self.transformer.cache_context("uncond"):
                         noise_uncond = self.transformer(
                             hidden_states=latents.to(transformer_dtype),
                             timestep=timestep,