Update logic to get display in scenario extractor

Fixes: 281084744
Test: atest FlickerLibTest
Change-Id: Ia94107bbaa74a9d71cd4b1ab03ac1e03371c6716
4 files changed