diff --git a/apps/CtsVerifier/src/com/android/cts/verifier/camera/its/ItsService.java b/apps/CtsVerifier/src/com/android/cts/verifier/camera/its/ItsService.java
index 8710096..2d2121a 100644
--- a/apps/CtsVerifier/src/com/android/cts/verifier/camera/its/ItsService.java
+++ b/apps/CtsVerifier/src/com/android/cts/verifier/camera/its/ItsService.java
@@ -331,7 +331,8 @@
                     mCameraListener, mCameraHandler);
             mCameraCharacteristics = mCameraManager.getCameraCharacteristics(
                     devices[cameraId]);
-            Size maxYuvSize = ItsUtils.getYuvOutputSizes(mCameraCharacteristics)[0];
+            Size maxYuvSize = ItsUtils.getMaxOutputSize(
+                    mCameraCharacteristics, ImageFormat.YUV_420_888);
             // 2 bytes per pixel for RGBA Bitmap and at least 3 Bitmaps per CDD
             int quota = maxYuvSize.getWidth() * maxYuvSize.getHeight() * 2 * 3;
             mSocketQueueQuota = new Semaphore(quota, true);