diff --git a/src/seismic_hazard_forecasting.py b/src/seismic_hazard_forecasting.py
index b45ebda..0a41da9 100644
--- a/src/seismic_hazard_forecasting.py
+++ b/src/seismic_hazard_forecasting.py
@@ -656,7 +656,7 @@ verbose: {verbose}")
 
                 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]
+                                                             IMT_max=2.0, rx_label=i, rtol=0.1, use_cython=True) for i in iter[:2]]
                 #imls = [dask.delayed(my_trivial_task)(i) for i in iter]
                 iml = dask.compute(*imls)
                 iml_grid_raw.append(list(iml))