diff --git a/nodes/sampling/ksampler_context.py b/nodes/sampling/ksampler_context.py
index b0dfe12..db1f072 100644
--- a/nodes/sampling/ksampler_context.py
+++ b/nodes/sampling/ksampler_context.py
@@ -40,5 +40,6 @@ class TT_KSamplerContext():
         }
 
         samples = sample_latents(**args)
+        context["latent"] = samples
 
         return (context, samples)