diff --git a/Dockerfile b/Dockerfile
index 8c7b2d9..fa49d02 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -3,7 +3,7 @@ FROM ubuntu:latest
 RUN apt-get update && \
     apt-get install -y python3-pip
 
-RUN pip3 install --user kaggle pandas scikit-learn sklearn
+RUN pip3 install --user kaggle pandas scikit-learn
 
 ENV PATH="/root/.local/bin:${PATH}"
 
diff --git a/data_processing.py b/data_processing.py
index adb9b42..c2eb2f4 100644
--- a/data_processing.py
+++ b/data_processing.py
@@ -1,6 +1,6 @@
-import sklearn
-from sklearn.preprocessing import OneHotEncoder
-from sklearn.model_selection import train_test_split
+import scikit-learn as sklearn
+from scikit-learn.preprocessing import OneHotEncoder
+from scikit-learn.model_selection import train_test_split
 import pandas as pd
 import subprocess