diff --git a/tft/module.py b/tft/module.py
index 119c824..96c3c34 100644
--- a/tft/module.py
+++ b/tft/module.py
@@ -579,9 +579,8 @@ class TFTModel(nn.Module):
             repeats=self.num_parallel_samples, dim=0
         )
 
-        repeated_past_target = (
-            past_target.repeat_interleave(repeats=self.num_parallel_samples, dim=0)
-            / repeated_scale
+        repeated_past_target = target.repeat_interleave(
+            repeats=self.num_parallel_samples, dim=0
         )
         repeated_past_selection = past_selection.repeat_interleave(
             repeats=self.num_parallel_samples, dim=0