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