diff --git a/graphstorm-processing/graphstorm_processing/data_transformations/dist_transformations/dist_hard_negative_transformation.py b/graphstorm-processing/graphstorm_processing/data_transformations/dist_transformations/dist_hard_negative_transformation.py
index ad7e26987c..083bcb1924 100755
--- a/graphstorm-processing/graphstorm_processing/data_transformations/dist_transformations/dist_hard_negative_transformation.py
+++ b/graphstorm-processing/graphstorm_processing/data_transformations/dist_transformations/dist_hard_negative_transformation.py
@@ -42,7 +42,7 @@ def apply_transform(
     spark: SparkSession
         The spark session
     input_df : DataFrame
-        The input DataFrame to apply normalization to.
+        The input DataFrame to apply transformation to.
     edge_mapping_dict: dict
         The mapping dictionary contain mapping file directory and edge type
     """