diff --git a/pts/trainer.py b/pts/trainer.py
index 25ddf73..9803f7e 100644
--- a/pts/trainer.py
+++ b/pts/trainer.py
@@ -50,8 +50,7 @@ class Trainer:
                     avg_epoch_loss += loss.item()
                     it.set_postfix(
                         ordered_dict={
-                            ("" if is_training else "validation_")
-                            + "avg_epoch_loss": avg_epoch_loss/batch_no
+                            "avg_epoch_loss": avg_epoch_loss/batch_no
                         },
                         refresh=False,
                     )