diff --git a/src/back/managing_missing_values.py b/src/back/managing_missing_values.py
index 24ac773..b12c463 100644
--- a/src/back/managing_missing_values.py
+++ b/src/back/managing_missing_values.py
@@ -48,7 +48,7 @@ def impute_with_regression(data):
         model = LinearRegression()
         model.fit(X_complete, y_complete)
         y_pred = model.predict(X_missing)
-        data.loc[df[col].isnull(), col] = y_pred
+        data.loc[data[col].isnull(), col] = y_pred
         
     return data
 
diff --git a/src/pages/clean.py b/src/pages/clean.py
index f64bd49..8fd0b97 100644
--- a/src/pages/clean.py
+++ b/src/pages/clean.py
@@ -13,7 +13,6 @@ if 'original_df' in st.session_state:
     st.write("## Missing data")
     rm_empty_rows_or_cols = st.checkbox("Remove empty rows or columns", True)
 
-
     st.write("#### Replace missing values")
     replace_methods = ["mean","median","mode","knn","regression"]
     replace_method = st.radio('Choose an option:', replace_methods)