diff --git a/pl_bolts/callbacks/vision/image_generation.py b/pl_bolts/callbacks/vision/image_generation.py
index 0859aa42e2..83ee748d05 100644
--- a/pl_bolts/callbacks/vision/image_generation.py
+++ b/pl_bolts/callbacks/vision/image_generation.py
@@ -90,6 +90,4 @@ def on_epoch_end(self, trainer: Trainer, pl_module: LightningModule) -> None:
             pad_value=self.pad_value,
         )
         str_title = f"{pl_module.__class__.__name__}_images"
-        trainer.logger.experiment.add_image(
-            str_title, grid, global_step=trainer.global_step
-        )
+        trainer.logger.experiment.add_image(str_title, grid, global_step=trainer.global_step)