diff --git a/deepinv_bench/benchmarks/plots/deepinv_boxplot.py b/deepinv_bench/base/plots.py
similarity index 97%
rename from deepinv_bench/benchmarks/plots/deepinv_boxplot.py
rename to deepinv_bench/base/plots.py
index 4e2479d..6cb7712 100644
--- a/deepinv_bench/benchmarks/plots/deepinv_boxplot.py
+++ b/deepinv_bench/base/plots.py
@@ -42,4 +42,4 @@ def plot(self, df, metric):
                 }
             )
 
-        return plot_data
+        return plot_data
\ No newline at end of file
diff --git a/deepinv_bench/benchmarks/benchmark_template/plots/deepinv_boxplot.py b/deepinv_bench/benchmarks/benchmark_template/plots/deepinv_boxplot.py
index 161079d..8e625eb 100644
--- a/deepinv_bench/benchmarks/benchmark_template/plots/deepinv_boxplot.py
+++ b/deepinv_bench/benchmarks/benchmark_template/plots/deepinv_boxplot.py
@@ -1,4 +1,4 @@
-from deepinv_bench.benchmarks.plots.deepinv_boxplot import Plot as _Plot
+from deepinv_bench.base.plots.deepinv_boxplot import Plot as _Plot
 
 
 class Plot(_Plot):
diff --git a/deepinv_bench/benchmarks/cbsd68_gaussian_denoising/plots/deepinv_boxplot.py b/deepinv_bench/benchmarks/cbsd68_gaussian_denoising/plots/deepinv_boxplot.py
index 161079d..8e625eb 100644
--- a/deepinv_bench/benchmarks/cbsd68_gaussian_denoising/plots/deepinv_boxplot.py
+++ b/deepinv_bench/benchmarks/cbsd68_gaussian_denoising/plots/deepinv_boxplot.py
@@ -1,4 +1,4 @@
-from deepinv_bench.benchmarks.plots.deepinv_boxplot import Plot as _Plot
+from deepinv_bench.base.plots.deepinv_boxplot import Plot as _Plot
 
 
 class Plot(_Plot):
diff --git a/deepinv_bench/benchmarks/div2k_gaussian_deblurring/plots/deepinv_boxplot.py b/deepinv_bench/benchmarks/div2k_gaussian_deblurring/plots/deepinv_boxplot.py
index 161079d..8e625eb 100644
--- a/deepinv_bench/benchmarks/div2k_gaussian_deblurring/plots/deepinv_boxplot.py
+++ b/deepinv_bench/benchmarks/div2k_gaussian_deblurring/plots/deepinv_boxplot.py
@@ -1,4 +1,4 @@
-from deepinv_bench.benchmarks.plots.deepinv_boxplot import Plot as _Plot
+from deepinv_bench.base.plots.deepinv_boxplot import Plot as _Plot
 
 
 class Plot(_Plot):
diff --git a/deepinv_bench/benchmarks/div2k_inpainting_easy/plots/deepinv_boxplot.py b/deepinv_bench/benchmarks/div2k_inpainting_easy/plots/deepinv_boxplot.py
index 161079d..8e625eb 100644
--- a/deepinv_bench/benchmarks/div2k_inpainting_easy/plots/deepinv_boxplot.py
+++ b/deepinv_bench/benchmarks/div2k_inpainting_easy/plots/deepinv_boxplot.py
@@ -1,4 +1,4 @@
-from deepinv_bench.benchmarks.plots.deepinv_boxplot import Plot as _Plot
+from deepinv_bench.base.plots.deepinv_boxplot import Plot as _Plot
 
 
 class Plot(_Plot):
diff --git a/deepinv_bench/benchmarks/div2k_super_resolution_2x/plots/deepinv_boxplot.py b/deepinv_bench/benchmarks/div2k_super_resolution_2x/plots/deepinv_boxplot.py
index 161079d..8e625eb 100644
--- a/deepinv_bench/benchmarks/div2k_super_resolution_2x/plots/deepinv_boxplot.py
+++ b/deepinv_bench/benchmarks/div2k_super_resolution_2x/plots/deepinv_boxplot.py
@@ -1,4 +1,4 @@
-from deepinv_bench.benchmarks.plots.deepinv_boxplot import Plot as _Plot
+from deepinv_bench.base.plots.deepinv_boxplot import Plot as _Plot
 
 
 class Plot(_Plot):