diff --git a/eynollah/eynollah/eynollah.py b/eynollah/eynollah/eynollah.py
index a68e26f..300fc53 100644
--- a/eynollah/eynollah/eynollah.py
+++ b/eynollah/eynollah/eynollah.py
@@ -31,7 +31,7 @@ warnings.filterwarnings("ignore")
 from scipy.signal import find_peaks
 import matplotlib.pyplot as plt
 from scipy.ndimage import gaussian_filter1d
-from tensorflow.python.keras.backend import set_session
+from tensorflow.compat.v1.keras.backend import set_session
 from tensorflow.keras import layers
 
 from .utils.contour import (