diff --git a/color_transfer.py b/color_transfer.py
index 14f4185..2f4b48e 100644
--- a/color_transfer.py
+++ b/color_transfer.py
@@ -29,7 +29,7 @@ def SwitchColors(image, detected_colors, target_colors, clustering_model, distan
     distance_methods = {
     "Euclidean": EuclideanDistance,
     "Manhattan": ManhattanDistance,
-    "HSV_Similarity": CosineSimilarity
+    "Cosine Similarity": CosineSimilarity
     }
 
     distance_method = distance_methods.get(distance_method)
@@ -55,7 +55,7 @@ class PaletteTransferNode:
                 "target_colors": ("COLOR_LIST",),
                 "color_space": (["RGB", "HSV", "LAB"], {'default': 'RGB'}),
                 "cluster_method": (["Kmeans","Mini batch Kmeans"], {'default': 'Kmeans'}, ),
-                "distance_method": (["Euclidean", "Manhattan", "CosineSimilarity"], {'default': 'Euclidean'}, )
+                "distance_method": (["Euclidean", "Manhattan", "Cosine Similarity"], {'default': 'Euclidean'}, )
                 }
             }
         return data_in