diff --git a/src/seismic_hazard_forecasting.py b/src/seismic_hazard_forecasting.py
index 2469d9d..a4d929a 100644
--- a/src/seismic_hazard_forecasting.py
+++ b/src/seismic_hazard_forecasting.py
@@ -81,7 +81,7 @@ def compute_IMT_exceedance(rx_lat, rx_lon, r, fr, p, lambdas, D, percentages_D,
                                     rake = Rake)
 
     ctx = context_maker.from_srcs([src], site_collection)[0] #returns one context from the source for one rupture   
-    return None
+
     if use_cython:
 
         from cython_exceedance import exceedance_core
@@ -146,7 +146,7 @@ def compute_IMT_exceedance(rx_lat, rx_lon, r, fr, p, lambdas, D, percentages_D,
         
             return exceedance_prob_sum - p
     
-
+    return None
     
     # Check function values at different test points
     IMT_mid = (IMT_max-IMT_min)/2