diff --git a/automotive/evs/aidl/vts/VtsHalEvsTargetTest.cpp b/automotive/evs/aidl/vts/VtsHalEvsTargetTest.cpp
index 60adf93..a6d99ad 100644
--- a/automotive/evs/aidl/vts/VtsHalEvsTargetTest.cpp
+++ b/automotive/evs/aidl/vts/VtsHalEvsTargetTest.cpp
@@ -2273,6 +2273,10 @@
             DisplayState state;
             EXPECT_FALSE(mEnumerator->getDisplayState(&state).isOk());
         }
+        for (const auto displayIdToQuery : displayIds) {
+            DisplayState state;
+            EXPECT_FALSE(mEnumerator->getDisplayStateById(displayIdToQuery, &state).isOk());
+        }
 
         // Scope to limit the lifetime of the pDisplay pointer, and thus the IEvsDisplay object.
         {
@@ -2285,6 +2289,15 @@
                 EXPECT_TRUE(mEnumerator->getDisplayState(&state).isOk());
                 EXPECT_EQ(state, DisplayState::NOT_VISIBLE);
             }
+            for (const auto displayIdToQuery : displayIds) {
+                DisplayState state;
+                if (displayIdToQuery == displayId) {
+                    EXPECT_TRUE(mEnumerator->getDisplayStateById(displayIdToQuery, &state).isOk());
+                    EXPECT_EQ(state, DisplayState::NOT_VISIBLE);
+                } else {
+                    EXPECT_FALSE(mEnumerator->getDisplayStateById(displayIdToQuery, &state).isOk());
+                }
+            }
 
             // Activate the display.
             EXPECT_TRUE(pDisplay->setDisplayState(DisplayState::VISIBLE_ON_NEXT_FRAME).isOk());
@@ -2298,6 +2311,15 @@
                 EXPECT_TRUE(pDisplay->getDisplayState(&state).isOk());
                 EXPECT_EQ(state, DisplayState::VISIBLE_ON_NEXT_FRAME);
             }
+            for (const auto displayIdToQuery : displayIds) {
+                DisplayState state;
+                if (displayIdToQuery == displayId) {
+                    EXPECT_TRUE(mEnumerator->getDisplayStateById(displayIdToQuery, &state).isOk());
+                    EXPECT_EQ(state, DisplayState::VISIBLE_ON_NEXT_FRAME);
+                } else {
+                    EXPECT_FALSE(mEnumerator->getDisplayStateById(displayIdToQuery, &state).isOk());
+                }
+            }
 
             // Get the output buffer we'd use to display the imagery.
             BufferDesc tgtBuffer;
@@ -2319,6 +2341,15 @@
                 EXPECT_TRUE(pDisplay->getDisplayState(&state).isOk());
                 EXPECT_EQ(state, DisplayState::VISIBLE);
             }
+            for (const auto displayIdToQuery : displayIds) {
+                DisplayState state;
+                if (displayIdToQuery == displayId) {
+                    EXPECT_TRUE(mEnumerator->getDisplayStateById(displayIdToQuery, &state).isOk());
+                    EXPECT_EQ(state, DisplayState::VISIBLE);
+                } else {
+                    EXPECT_FALSE(mEnumerator->getDisplayStateById(displayIdToQuery, &state).isOk());
+                }
+            }
 
             // Turn off the display.
             EXPECT_TRUE(pDisplay->setDisplayState(DisplayState::NOT_VISIBLE).isOk());
@@ -2333,6 +2364,15 @@
                 EXPECT_TRUE(pDisplay->getDisplayState(&state).isOk());
                 EXPECT_EQ(state, DisplayState::NOT_VISIBLE);
             }
+            for (const auto displayIdToQuery : displayIds) {
+                DisplayState state;
+                if (displayIdToQuery == displayId) {
+                    EXPECT_TRUE(mEnumerator->getDisplayStateById(displayIdToQuery, &state).isOk());
+                    EXPECT_EQ(state, DisplayState::NOT_VISIBLE);
+                } else {
+                    EXPECT_FALSE(mEnumerator->getDisplayStateById(displayIdToQuery, &state).isOk());
+                }
+            }
 
             // Close the display.
             mEnumerator->closeDisplay(pDisplay);
@@ -2348,6 +2388,10 @@
             DisplayState state;
             EXPECT_FALSE(mEnumerator->getDisplayState(&state).isOk());
         }
+        for (const auto displayIdToQuery : displayIds) {
+            DisplayState state;
+            EXPECT_FALSE(mEnumerator->getDisplayStateById(displayIdToQuery, &state).isOk());
+        }
     }
 }