diff --git a/src/super_gradients/training/datasets/detection_datasets/detection_dataset.py b/src/super_gradients/training/datasets/detection_datasets/detection_dataset.py
index ff922e2c40..f14fafbf27 100644
--- a/src/super_gradients/training/datasets/detection_datasets/detection_dataset.py
+++ b/src/super_gradients/training/datasets/detection_datasets/detection_dataset.py
@@ -493,7 +493,7 @@ def plot(
 
             # Plot `max_samples_per_plot` images.
             for img_i in range(max_samples_per_plot):
-                index = img_i + plot_i * 16
+                index = img_i + plot_i * max_samples_per_plot
 
                 # LOAD IMAGE/TARGETS
                 if plot_transformed_data:
@@ -518,7 +518,7 @@ def plot(
 
                 image = DetectionVisualization.visualize_image(image_np=image, class_names=self.classes, target_boxes=targets_label_xyxy, gt_alpha=1)
 
-                plt.subplot(n_subplot, n_subplot, img_i + 1).imshow(image[:, :, ::-1])
+                plt.subplot(n_subplot, n_subplot, img_i + 1).imshow(image)
                 plt.imshow(image)
                 plt.axis("off")