Get Context for corner radius from correct display

Test: local + presubmit
Flag: EXEMPT bugfix
Bug: 416287239

Change-Id: If788e737f93e22e37bbc1a1aed1120119678aa97
4 files changed