diff --git a/src/defense/models.py b/src/defense/models.py
index cbb9224..ea6d2ca 100644
--- a/src/defense/models.py
+++ b/src/defense/models.py
@@ -169,13 +169,13 @@ def custom_callbacks(self) -> List[keras.callbacks.Callback]:
             y = defense_model.predict(x)
 
             tf.summary.image(
-                f"(Defense) Original images",
+                f"(Defense)[{args.dataset}] Original images",
                 concat_batch_images(x),
                 step=idx,
             )
 
             tf.summary.image(
-                f"(Defense) {args.defense.upper()} processing result",
+                f"(Defense)[{args.dataset}] {args.defense.upper()} processing result",
                 concat_batch_images(y),
                 step=idx,
             )