diff --git a/main.py b/main.py
index 260f293..cdf3f7c 100644
--- a/main.py
+++ b/main.py
@@ -118,8 +118,8 @@ def split_data(df, split_value):
 
     # Standardize features by removing the mean and scaling to unit variance.
     scale = StandardScaler()
-    x_train = scale.fit_transform(x_train)
-    x_test = scale.transform(x_test) 
+    X_train = scale.fit_transform(X_train)
+    X_test = scale.transform(X_test) 
 
     return X_train, X_test, y_train, y_test