diff --git a/Dockerfile b/Dockerfile
index d9a672d..8e3a8d9 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,7 +2,7 @@ FROM ubuntu:latest
 
 RUN apt-get update && apt-get install -y python3-pip unzip coreutils
 
-RUN pip install --no-cache-dir wheel kaggle pandas scikit-learn tensorflow
+RUN pip install --no-cache-dir wheel kaggle pandas scikit-learn tensorflow keras
 
 WORKDIR /app
 
diff --git a/predict.py b/predict.py
index 0cd688f..af76db4 100644
--- a/predict.py
+++ b/predict.py
@@ -4,6 +4,8 @@ from sklearn.preprocessing import StandardScaler, OneHotEncoder
 from sklearn.compose import ColumnTransformer
 from sklearn.pipeline import Pipeline
 from sklearn.model_selection import train_test_split
+from keras.metrics import MeanSquaredError
+
 
 loaded_model = tf.keras.models.load_model('powerlifting_model.h5')