diff --git a/src/seismic_hazard_forecasting.py b/src/seismic_hazard_forecasting.py
index be6721f..bb60f42 100644
--- a/src/seismic_hazard_forecasting.py
+++ b/src/seismic_hazard_forecasting.py
@@ -2,7 +2,7 @@
 import numpy as np
 import pandas as pd
 import Rbeast as rb
-from igfash.rate import datenum_to_datetime
+from igfash.rate import datenum_to_datetime, av_rates
 
 def apply_beast(act_rate, **kwargs):
     """