diff --git a/function.py b/function.py
index bf7067d..078b00a 100644
--- a/function.py
+++ b/function.py
@@ -126,4 +126,4 @@ if __name__ == "__main__":
 
     train_data, train_labels = load_data(train_file)
 
-    print(classify0([65915, 3.868359, 0.128078], train_data, train_labels))
+    print(classify([65915, 3.868359, 0.128078], train_data, train_labels))