1 files changed, 2 insertions, 2 deletions

diff --git a/drivers/media/video/zr364xx.c b/drivers/media/video/zr364xx.c
index 3d4bac252902..a82b5bd18d26 100644
--- a/drivers/media/video/zr364xx.c
+++ b/drivers/media/video/zr364xx.c
@@ -376,8 +376,8 @@ static int buffer_setup(struct videobuf_queue *vq, unsigned int *count,
         if (*count == 0)
                 *count = ZR364XX_DEF_BUFS;
 
-        while (*size * (*count) > ZR364XX_DEF_BUFS * 1024 * 1024)
-                (*count)--;
+        if (*size * *count > ZR364XX_DEF_BUFS * 1024 * 1024)
+                *count = (ZR364XX_DEF_BUFS * 1024 * 1024) / *size;
 
         return 0;
 }