diff --git a/pyproject.toml b/pyproject.toml
index 15c748de..dfe0fe55 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "shimmer"
-version = "0.1.0"
+version = "0.1.1"
 description = "A light GLoW"
 authors = ["bdvllrs <bdvllrs@gmail.com>"]
 license = "MIT"
diff --git a/shimmer/modules/losses.py b/shimmer/modules/losses.py
index 1a4944bc..78152d15 100644
--- a/shimmer/modules/losses.py
+++ b/shimmer/modules/losses.py
@@ -134,6 +134,7 @@ def _translation_loss(
                     }
                 )
     losses["translations"] = torch.stack(list(losses.values()), dim=0).mean()
+    losses.update(metrics)
     return losses