diff --git a/scripts/core/txt2vid.py b/scripts/core/txt2vid.py
index eda4b71..dc23760 100644
--- a/scripts/core/txt2vid.py
+++ b/scripts/core/txt2vid.py
@@ -87,7 +87,7 @@ def start_process(*args):
       utils.set_CNs_input_image(args_dict, Image.fromarray(curr_video_frame))
 
     processed_frames, _, _, _ = utils.txt2img(args_dict)
-    processed_frame = np.array(processed_frames[0])
+    processed_frame = np.array(processed_frames[0])[...,:3]
     processed_frame = np.clip(processed_frame, 0, 255).astype(np.uint8)
     init_frame = processed_frame.copy()
 
@@ -153,7 +153,7 @@ def start_process(*args):
         utils.set_CNs_input_image(args_dict, Image.fromarray(curr_video_frame))
 
       processed_frames, _, _, _ = utils.img2img(args_dict)
-      processed_frame = np.array(processed_frames[0])
+      processed_frame = np.array(processed_frames[0])[...,:3]
       processed_frame = skimage.exposure.match_histograms(processed_frame, init_frame, channel_axis=-1)
       processed_frame = np.clip(processed_frame, 0, 255).astype(np.uint8)