diff --git a/eynollah/eynollah/train/train.py b/eynollah/eynollah/train/train.py
index 0881182..ec6900c 100644
--- a/eynollah/eynollah/train/train.py
+++ b/eynollah/eynollah/train/train.py
@@ -1,7 +1,7 @@
 import os
 import sys
 import tensorflow as tf
-from keras.backend.tensorflow_backend import set_session
+from tensorflow.compat.v1.keras.backend import set_session
 import keras, warnings
 from keras.optimizers import *
 from sacred import Experiment