diff --git a/training/trainer.py b/training/trainer.py
index c41e9b6..5d956b1 100644
--- a/training/trainer.py
+++ b/training/trainer.py
@@ -143,7 +143,7 @@ def test_iteration(self):
 
         self.log.update("loss_val_Deformation_ChamferL2", loss_val_Deformation_ChamferL2)
         print(
-            '\r' + colored('[%d: %d/%d]' % (self.epoch, self.iteration, self.len_dataset_test / (self.opt.batch_size)),
+            '\r' + colored('[%d: %d/%d]' % (self.epoch, self.iteration, len(self.dataloader_test)),
                            'red') +
             colored('loss_val_Deformation_ChamferL2:  %f' % loss_val_Deformation_ChamferL2.item(), 'yellow'),
             end='')