diff --git a/style_transfer/style_transfer.py b/style_transfer/style_transfer.py
index 51226db..3365996 100644
--- a/style_transfer/style_transfer.py
+++ b/style_transfer/style_transfer.py
@@ -60,7 +60,7 @@ class StyleTransfer():
             stylized_batch_of_imgs = comixGAN_model.predict(batch_of_frames)
             stylized_imgs.append(255*stylized_batch_of_imgs)
 
-        return list(np.stack(stylized_imgs))
+        return list(np.concatenate(stylized_imgs, axis=0))
 
     @classmethod
     def _cartoon_gan_stylize(cls, frames, gpu=True, style='Hayao'):