diff --git a/data_processing.py b/data_processing.py
index 480b6c2..0c6be54 100644
--- a/data_processing.py
+++ b/data_processing.py
@@ -1,5 +1,6 @@
 import sklearn
 from sklearn.model_selection import train_test_split
+from sklearn.preprocessing import OneHotEncoder
 import pandas as pd
 
 housing_price_dataset = pd.read_csv('housing_price_dataset.csv')