diff --git a/function.py b/function.py
index b7951af..dc1be7d 100644
--- a/function.py
+++ b/function.py
@@ -139,7 +139,7 @@ if __name__ == "__main__":
         "00000000000001111111110000000000"
         "00000000000000000111100000000000"
     )
-    predicted_label = classify(
+    predicted_label = classify_mahalanobis(
         np.array([int(char) for char in test_data]).reshape(1, 1024),
         train_data,
         train_labels,