diff --git a/barlow_twin_like/main.py b/barlow_twin_like/main.py
index 2542c308a697f554e67e36a3e4a8631222fb67ba..01effe173511771c7f8650c1583e1892be2cd407 100644
--- a/barlow_twin_like/main.py
+++ b/barlow_twin_like/main.py
@@ -61,7 +61,7 @@ def test_representation(model, data_val, epoch, wandb):
     print('Val epoch {}, loss : {:.3f}'.format(epoch, losses))
 
     if wandb is not None:
-        wdb.log({"train loss": losses, 'train epoch': epoch})
+        wdb.log({"val loss": losses, 'val epoch': epoch})
 
     return losses