diff --git a/src/models/ocr_model/utils/ocr_aug.py b/src/models/ocr_model/utils/ocr_aug.py
index c3e1d12..5678c61 100644
--- a/src/models/ocr_model/utils/ocr_aug.py
+++ b/src/models/ocr_model/utils/ocr_aug.py
@@ -3,12 +3,11 @@ import random
 
 def ocr_augmentation_pipeline():
     pre_phase = [
-        # Rescale(scale="optimal", target_dpi = 300,  p = 1.0),
     ]
 
     ink_phase = [
         InkColorSwap(
-            ink_swap_color="lhy_custom",
+            ink_swap_color="random",
             ink_swap_sequence_number_range=(5, 10),
             ink_swap_min_width_range=(2, 3),
             ink_swap_max_width_range=(100, 120),