diff --git a/optimus/ml/keycollision.py b/optimus/ml/keycollision.py
index a1d9c57f..e2830abd 100644
--- a/optimus/ml/keycollision.py
+++ b/optimus/ml/keycollision.py
@@ -89,15 +89,14 @@ def calculate_ngrams(value, args):
 
 
 def fingerprint_cluster(df, input_cols, output: str = "dict"):
-    return base_cluster(df, input_cols, output, func=fingerprint, args=[input_cols])
+    return base_clustering_function(df, input_cols, output, func=fingerprint, args=[input_cols])
 
 
 def n_gram_fingerprint_cluster(df, input_cols, n_size=2, output: str = "dict"):
-    return base_cluster(df, input_cols, output, func=n_gram_fingerprint, args=[input_cols, n_size])
+    return base_clustering_function(df, input_cols, output, func=n_gram_fingerprint, args=[input_cols, n_size])
 
 
-# df, input_cols, output: str = "dict"
-def base_cluster(df, input_cols, output, func=None, args=None):
+def base_clustering_function(df, input_cols, output, func=None, args=None):
     """
     Cluster a dataframe column based on the Fingerprint algorithm
     :return: