diff --git a/eval.py b/eval.py
index 2ff6cd6..79738e3 100644
--- a/eval.py
+++ b/eval.py
@@ -113,8 +113,8 @@
     target[:, :3] = (target[:, :3] * pose_s) + pose_m
 
     # take the middle prediction
-    pred_poses[idx, :] = output[len(output) / 2]
-    targ_poses[idx, :] = target[len(target) / 2]
+    pred_poses[idx, :] = output
+    targ_poses[idx, :] = target
 
 # calculate losses
 t_loss = np.asarray([t_criterion(p, t) for p, t in zip(pred_poses[:, :3], targ_poses[:, :3])])
@@ -133,4 +133,4 @@
 plt.plot(gt_pose[0, 1], gt_pose[0, 0], 'y*', markersize=15)
 plt.show(block=True)
 image_filename = osp.join(osp.expanduser(opt.results_dir), '{:s}.png'.format(opt.exp_name))
-fig.savefig(image_filename)
\ No newline at end of file
+fig.savefig(image_filename)