diff --git a/lerobot/common/policies/smolvla/modeling_smolvla.py b/lerobot/common/policies/smolvla/modeling_smolvla.py
index efdf602d4..008ba8380 100644
--- a/lerobot/common/policies/smolvla/modeling_smolvla.py
+++ b/lerobot/common/policies/smolvla/modeling_smolvla.py
@@ -338,7 +338,7 @@ class SmolVLAPolicy(PreTrainedPolicy):
         # For backward pass
         loss = losses.mean()
         # For backward pass
-        loss_dict["loss"] = loss
+        loss_dict["loss"] = loss.item()
         return loss, loss_dict
 
     def prepare_images(self, batch):
diff --git a/lerobot/common/utils/wandb_utils.py b/lerobot/common/utils/wandb_utils.py
index 3fe241d41..9e938e191 100644
--- a/lerobot/common/utils/wandb_utils.py
+++ b/lerobot/common/utils/wandb_utils.py
@@ -115,7 +115,7 @@ class WandBLogger:
         for k, v in d.items():
             if not isinstance(v, (int, float, str)):
                 logging.warning(
-                    f'WandB logging of key "{k}" was ignored as its type is not handled by this wrapper.'
+                    f'WandB logging of key "{k}" was ignored as its type "{type(v)}" is not handled by this wrapper.'
                 )
                 continue
             self._wandb.log({f"{mode}/{k}": v}, step=step)