diff --git a/core/imagelib/color_transfer.py b/core/imagelib/color_transfer.py
index 185dac4..927dc9f 100644
--- a/core/imagelib/color_transfer.py
+++ b/core/imagelib/color_transfer.py
@@ -373,8 +373,8 @@ def color_transfer(ct_mode, img_src, img_trg):
 # imported from faceswap
 def color_augmentation(img, seed=None):
     """ Color adjust RGB image """
-    face = img
-    face = np.clip(face*255.0, 0, 255).astype(np.uint8)
+    face = img.astype(np.float32)
+    face = np.clip(face*255.0, 0, 255)
     face = random_clahe(face, seed)
     face = random_lab(face, seed)
     img[:, :, :3] = face