diff --git a/data_process.py b/data_process.py
index 9536c7e..5b5bf89 100644
--- a/data_process.py
+++ b/data_process.py
@@ -84,7 +84,7 @@ def CutOff(self, distance, max_id, threshold):
             
     def Guasse(self, distance, max_id, threshold):
         '''
-        :rtype: dict with Gaussian kernel values by desc
+        :rtype: list with Gaussian kernel values by desc
         '''
         guasse = dict()
         for i in range(1, max_id + 1):
@@ -151,4 +151,4 @@ def show_pair_info(self, pair, threshold):
         plt.ylabel(r'$\delta$')
         plt.title(r'$d_c=$' + str(threshold))
         plt.savefig('./images/Decision Graph Cutoff.png')
-        plt.close()
\ No newline at end of file
+        plt.close()