diff --git a/drivers/media/platform/rockchip/vpss/vpss_rockit.c b/drivers/media/platform/rockchip/vpss/vpss_rockit.c
index 5d1f831cb318..f419bcfa358b 100644
--- a/drivers/media/platform/rockchip/vpss/vpss_rockit.c
+++ b/drivers/media/platform/rockchip/vpss/vpss_rockit.c
@@ -73,6 +73,12 @@ static struct rkvpss_stream *rkvpss_rockit_get_stream(struct rockit_rkvpss_cfg *
 	case 3:
 		stream = &vpss_dev->stream_vdev.stream[RKVPSS_OUTPUT_CH3];
 		break;
+	case 4:
+		stream = &vpss_dev->stream_vdev.stream[RKVPSS_OUTPUT_CH4];
+		break;
+	case 5:
+		stream = &vpss_dev->stream_vdev.stream[RKVPSS_OUTPUT_CH5];
+		break;
 	default:
 		stream = NULL;
 		break;