diff --git a/.ml/.azure/run.json b/.ml/.azure/run.json
index 4b79e7c..e43e730 100644
--- a/.ml/.azure/run.json
+++ b/.ml/.azure/run.json
@@ -1,3 +1,3 @@
 {
-    "experiment": "GitHubActionExperiment"
-}
\ No newline at end of file
+    "experiment_name": "GitHubActionExperiment"
+}
diff --git a/src/train/train.py b/src/train/train.py
index 1540be5..8e70436 100644
--- a/src/train/train.py
+++ b/src/train/train.py
@@ -78,7 +78,7 @@ def main(args):
     X, y = datasets.load_iris(return_X_y=True)
     
     #dividing X,y into train and test data
-    x_train, x_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=223)
+    x_train, x_test, y_train, y_test = train_test_split(X, y, test_size=0.23, random_state=223)
     data = {'train': {'X': x_train, 'y': y_train},
             'test': {'X': x_test, 'y': y_test}}