diff --git a/src/pipelines/pipeline_pose2vid_long.py b/src/pipelines/pipeline_pose2vid_long.py
index 1a79410..7323a77 100644
--- a/src/pipelines/pipeline_pose2vid_long.py
+++ b/src/pipelines/pipeline_pose2vid_long.py
@@ -475,18 +475,6 @@ def __call__(
                     )
                     reference_control_reader.update(reference_control_writer)
 
-                context_queue = list(
-                    context_scheduler(
-                        0,
-                        num_inference_steps,
-                        latents.shape[2],
-                        context_frames,
-                        context_stride,
-                        0,
-                    )
-                )
-                num_context_batches = math.ceil(len(context_queue) / context_batch_size)
-
                 context_queue = list(
                     context_scheduler(
                         0,
@@ -499,6 +487,7 @@ def __call__(
                 )
 
                 num_context_batches = math.ceil(len(context_queue) / context_batch_size)
+
                 global_context = []
                 for i in range(num_context_batches):
                     global_context.append(