diff --git a/machine_learning/lstm/lstm_prediction.py_tf b/machine_learning/lstm/lstm_prediction.py
similarity index 98%
rename from machine_learning/lstm/lstm_prediction.py_tf
rename to machine_learning/lstm/lstm_prediction.py
index 5452f0443f62..b9a26b8f384b 100644
--- a/machine_learning/lstm/lstm_prediction.py_tf
+++ b/machine_learning/lstm/lstm_prediction.py
@@ -6,6 +6,7 @@
 """
 import numpy as np
 import pandas as pd
+from tensorflow import keras
 from keras.layers import LSTM, Dense
 from keras.models import Sequential
 from sklearn.preprocessing import MinMaxScaler