diff --git a/knn.py b/knn.py
index 2e117427f5124a19a11cdfa1c16bbd040f45ad77..2edab7c1e1dec71f2b6e0151146193e3387e0480 100644
--- a/knn.py
+++ b/knn.py
@@ -57,7 +57,7 @@ if __name__ == "__main__":
     k_list = [k for k in range(1, 21)]
     accuracy = [evaluate_knn(data_train, labels_train, data_test, labels_test, k) for k in range (1, 21)]
     
-    plt.plot([k for k in range (1, 21)], accuracy)
+    plt.plot(k_list, accuracy)
     plt.title("Variation of k-nearest neighbors method accuracy for k from 1 to 20")
     plt.xlabel("k value")
     plt.ylabel("Accuracy")