diff --git a/knn.py b/knn.py
index a7decbb1606c31b55c1a0d9f66d2cc88b4665667..8b7450e844c4d9d81abbfbcdf73ec667e20a0fd1 100644
--- a/knn.py
+++ b/knn.py
@@ -49,7 +49,7 @@ if __name__ == "__main__":
     data, labels = read_cifar("./data/cifar-10-batches-py")
     
     # Formatting the data into training and testing sets
-    data_train, labels_train, data_test, labels_test = split_dataset(data, labels, 0.1)
+    data_train, labels_train, data_test, labels_test = split_dataset(data, labels, 0.9)
     
     # Data to plot
     k_list = [k for k in range(1, 21)]