diff --git a/ehrapy/plot/_survival_analysis.py b/ehrapy/plot/_survival_analysis.py
index 444f0be2..b02caeb6 100644
--- a/ehrapy/plot/_survival_analysis.py
+++ b/ehrapy/plot/_survival_analysis.py
@@ -8,7 +8,6 @@
 import matplotlib.ticker as ticker
 import numpy as np
 import pandas as pd
-from matplotlib import gridspec
 from numpy import ndarray
 
 from ehrapy.plot import scatter