diff --git a/src/ML_main.py b/src/ML_main.py
index aeb11ac..9abb0c0 100644
--- a/src/ML_main.py
+++ b/src/ML_main.py
@@ -167,11 +167,11 @@ def plot_t_sne(dataset):
    dataset = np.loadtxt(dataset_file)
    label = np.loadtxt(label_file)
     
-   selected_dataset_file = path + r"\dataset.txt" 
-   selected_label_file = path + r"\label.txt" 
+   selected_dataset_file = path + r"\dataset_selected.txt" 
+   selected_label_file = path + r"\label_selected.txt" 
    we_selected_dataset = np.loadtxt(selected_dataset_file)
    we_selected_label = np.loadtxt(selected_label_file)
-    
+   
 
    # load dataset with labels
    both_file = path + r"\both.txt"