diff --git a/sdm/libs/hwc/hwc_session.cpp b/sdm/libs/hwc/hwc_session.cpp
index 9393b81..dbd06a7 100644
--- a/sdm/libs/hwc/hwc_session.cpp
+++ b/sdm/libs/hwc/hwc_session.cpp
@@ -232,7 +232,7 @@
                         hwc_display_contents_1_t **displays) {
   DTRACE_SCOPED();
 
-  if (!device || !displays) {
+  if (!device || !displays || num_displays > HWC_NUM_DISPLAY_TYPES) {
     return -EINVAL;
   }
 
@@ -312,7 +312,7 @@
 
   SEQUENCE_EXIT_SCOPE_LOCK(locker_);
 
-  if (!device || !displays) {
+  if (!device || !displays || num_displays > HWC_NUM_DISPLAY_TYPES) {
     return -EINVAL;
   }