diff --git a/src/seismic_hazard_forecasting.py b/src/seismic_hazard_forecasting.py
index f8c8afd..96932d8 100644
--- a/src/seismic_hazard_forecasting.py
+++ b/src/seismic_hazard_forecasting.py
@@ -469,11 +469,10 @@ verbose: {verbose}")
             for imt in products:
                 logger.info(f"Estimating {imt}")
 
-                #imls = [dask.delayed(compute_IMT_exceedance)(rx_lat[i], rx_lon[i], distances[i].flatten(), fr, p, lambdas,
-                #                                            forecast_len, lambdas_perc, m_range, m_pdf, m_cdf, model,
-                #                                            log_level=logging.DEBUG, imt=imt, IMT_min=0.0, IMT_max=2.0, rx_label=i,
-                #                                            rtol=0.1, use_cython=True) for i in iter]
-                imls = [dask.delayed(my_trivial_task)(i) for i in iter]
+                imls = [dask.delayed(compute_IMT_exceedance)(rx_lat[i], rx_lon[i], distances[i].flatten(), fr, p, lambdas, forecast_len,
+                                                             lambdas_perc, m_range, m_pdf, m_cdf, model, imt=imt, IMT_min=0.0,
+                                                             IMT_max=2.0, rx_label=i, rtol=0.1, use_cython=True) for i in iter]
+                #imls = [dask.delayed(my_trivial_task)(i) for i in iter]
                 iml = dask.compute(*imls)
                 iml_grid_raw.append(list(iml))