diff --git a/mlair/run_modules/post_processing.py b/mlair/run_modules/post_processing.py
index 6acfcfbb9a16a6a5af37afb6b3c56807f8f2ff5c..c5f5b2d3084f707eb46bb59e880c9579d2cbbbf7 100644
--- a/mlair/run_modules/post_processing.py
+++ b/mlair/run_modules/post_processing.py
@@ -309,13 +309,11 @@ class PostProcessing(RunEnvironment):
         try:
             if (self.data_store.get('oversampling_method')=='bin_oversampling') and (
                     "PlotOversamplingContingency" in plot_list):
-                bin_edges = self.data_store.get('oversampling_bin_edges')
-                oversampling_rates = self.data_store.get('oversampling_rates_capped','train')
                 predictions = None
                 labels = None
-                PlotOversampling(predictions, labels, plot_folder=self.plot_path)
+                PlotOversamplingContingency(predictions, labels, plot_folder=self.plot_path)
         except Exception as e:
-            logging.error(f"Could not create plot OversamplingPlots due to the following error: {e}")
+            logging.error(f"Could not create plot OversamplingContingencyPlots due to the following error: {e}")
 
         try:
             if (self.data_store.get('oversampling_method')=='bin_oversampling') and (