diff --git a/pokemonred_puffer/cleanrl_puffer.py b/pokemonred_puffer/cleanrl_puffer.py
index aa6e3fe..c8230a3 100644
--- a/pokemonred_puffer/cleanrl_puffer.py
+++ b/pokemonred_puffer/cleanrl_puffer.py
@@ -309,7 +309,9 @@ def evaluate(self):
                             np.stack(self.infos["pokemon_exploration_map"], axis=0)
                         )
                         if self.wandb_client is not None:
-                            self.stats["Media/aggregate_exploration_map"] = wandb.Image(overlay)
+                            self.stats["Media/aggregate_exploration_map"] = wandb.Image(
+                                overlay, file_type="jpg"
+                            )
                     elif "state" in k:
                         continue