diff --git a/machine_learning/lstm/lstm_prediction.py_tf b/machine_learning/lstm/lstm_prediction.py
similarity index 95%
rename from machine_learning/lstm/lstm_prediction.py_tf
rename to machine_learning/lstm/lstm_prediction.py
index 5452f0443..6fd3cf291 100644
--- a/machine_learning/lstm/lstm_prediction.py_tf
+++ b/machine_learning/lstm/lstm_prediction.py
@@ -6,9 +6,9 @@
 """
 import numpy as np
 import pandas as pd
-from keras.layers import LSTM, Dense
-from keras.models import Sequential
 from sklearn.preprocessing import MinMaxScaler
+from tensorflow.keras.layers import LSTM, Dense
+from tensorflow.keras.models import Sequential
 
 if __name__ == "__main__":
     """