diff --git a/function.py b/function.py
index 078b00a..8b980c6 100644
--- a/function.py
+++ b/function.py
@@ -124,6 +124,6 @@ if __name__ == "__main__":
 
     train_file = "datingTrainSet2.tsv"
 
-    train_data, train_labels = load_data(train_file)
+    train_data, train_labels = load_dataset(train_file)
 
     print(classify([65915, 3.868359, 0.128078], train_data, train_labels))