diff --git a/deforum-stable-diffusion/helpers/colors.py b/deforum-stable-diffusion/helpers/colors.py
index 6ec81e1..1dec761 100644
--- a/deforum-stable-diffusion/helpers/colors.py
+++ b/deforum-stable-diffusion/helpers/colors.py
@@ -3,14 +3,14 @@
 
 def maintain_colors(prev_img, color_match_sample, mode):
     if mode == 'Match Frame 0 RGB':
-        return match_histograms(prev_img, color_match_sample, multichannel=True)
+        return match_histograms(prev_img, color_match_sample, channel_axis=-1)
     elif mode == 'Match Frame 0 HSV':
         prev_img_hsv = cv2.cvtColor(prev_img, cv2.COLOR_RGB2HSV)
         color_match_hsv = cv2.cvtColor(color_match_sample, cv2.COLOR_RGB2HSV)
-        matched_hsv = match_histograms(prev_img_hsv, color_match_hsv, multichannel=True)
+        matched_hsv = match_histograms(prev_img_hsv, color_match_hsv, channel_axis=-1)
         return cv2.cvtColor(matched_hsv, cv2.COLOR_HSV2RGB)
     else: # Match Frame 0 LAB
         prev_img_lab = cv2.cvtColor(prev_img, cv2.COLOR_RGB2LAB)
         color_match_lab = cv2.cvtColor(color_match_sample, cv2.COLOR_RGB2LAB)
-        matched_lab = match_histograms(prev_img_lab, color_match_lab, multichannel=True)
-        return cv2.cvtColor(matched_lab, cv2.COLOR_LAB2RGB)
\ No newline at end of file
+        matched_lab = match_histograms(prev_img_lab, color_match_lab, channel_axis=-1)
+        return cv2.cvtColor(matched_lab, cv2.COLOR_LAB2RGB)