diff --git a/aum/aum.py b/aum/aum.py
index f23c6f2..d83242a 100644
--- a/aum/aum.py
+++ b/aum/aum.py
@@ -83,10 +83,10 @@ def update(self, logits: torch.Tensor, targets: torch.Tensor,
 
             record = AUMRecord(sample_id=sample_id,
                                num_measurements=self.counts[sample_id],
-                               target_logit=targets[i].item(),
-                               target_val=target_values[i].item(),
-                               other_logit=other_logit_index[i].item(),
-                               other_val=other_logit_values[i].item(),
+                               target_logit=target_values[i].item(),
+                               target_val=targets[i].item(),
+                               other_logit=other_logit_values[i].item(),
+                               other_val=other_logit_index[i].item(),
                                margin=margin,
                                aum=self.sums[sample_id] / self.counts[sample_id])