diff --git a/pts/model/causal_deepar/causal_deepar_network.py b/pts/model/causal_deepar/causal_deepar_network.py
index 64a9c2c..ff1e787 100644
--- a/pts/model/causal_deepar/causal_deepar_network.py
+++ b/pts/model/causal_deepar/causal_deepar_network.py
@@ -177,8 +177,8 @@ class CausalDeepARNetwork(nn.Module):
         # scale is computed on the context length last units of the past target
         # scale shape is (batch_size, 1, *target_shape)
         _, scale = self.scaler(
-            past_target[:, self.context_length :, ...],
-            past_observed_values[:, self.context_length :, ...],
+            past_target[:, -self.context_length :, ...],
+            past_observed_values[:, -self.context_length :, ...],
         )
 
         _, control_scale = self.control_scaler(
diff --git a/pts/model/deepar/deepar_network.py b/pts/model/deepar/deepar_network.py
index bd98e05..12f1709 100644
--- a/pts/model/deepar/deepar_network.py
+++ b/pts/model/deepar/deepar_network.py
@@ -160,8 +160,8 @@ class DeepARNetwork(nn.Module):
         # scale is computed on the context length last units of the past target
         # scale shape is (batch_size, 1, *target_shape)
         _, scale = self.scaler(
-            past_target[:, self.context_length :, ...],
-            past_observed_values[:, self.context_length :, ...],
+            past_target[:, -self.context_length :, ...],
+            past_observed_values[:, -self.context_length :, ...],
         )
 
         # (batch_size, num_features)