diff --git a/src/seismic_hazard_forecasting.py b/src/seismic_hazard_forecasting.py
index 67db6da..3b9fa7a 100644
--- a/src/seismic_hazard_forecasting.py
+++ b/src/seismic_hazard_forecasting.py
@@ -3,6 +3,7 @@ from eqdist.rate import datenum_to_datetime
 import Rbeast as rb;
 from scipy.stats import bootstrap
 from matplotlib.dates import DateFormatter, AutoDateLocator
+import matplotlib.pyplot as plt
 import numpy as np
 
 global ncp_choice, tcp_max, torder_min, torder_max, AOI_lat, AOI_lon