diff --git a/app/src/main/java/ru/gordinmitya/yuv2buf_demo/RenderScriptConverter.kt b/app/src/main/java/ru/gordinmitya/yuv2buf_demo/RenderScriptConverter.kt
index 9d1e7cc..b5ccf57 100644
--- a/app/src/main/java/ru/gordinmitya/yuv2buf_demo/RenderScriptConverter.kt
+++ b/app/src/main/java/ru/gordinmitya/yuv2buf_demo/RenderScriptConverter.kt
@@ -34,10 +34,14 @@ class RenderScriptConverter(context: Context) : ImageConverter {
             || input!!.type.yuv != converted.type.format
             || bytes.size != converted.buffer.capacity()
         ) {
+            val yuvFormat = when (converted.type) {
+                Yuv.Type.YUV_I420 -> ImageFormat.YUV_420_888
+                Yuv.Type.YUV_NV21 -> ImageFormat.NV21
+            }
             val yuvType: Type.Builder = Type.Builder(rs, Element.U8(rs))
                 .setX(image.width)
                 .setY(image.height)
-                .setYuvFormat(ImageFormat.NV21)
+                .setYuvFormat(yuvFormat)
             input = Allocation.createTyped(rs, yuvType.create(), Allocation.USAGE_SCRIPT)
             bytes = ByteArray(converted.buffer.capacity())
             val rgbaType: Type.Builder = Type.Builder(rs, Element.RGBA_8888(rs))