Update when display change callback on legacy transition

This callback is only used on shell transtion before, but we should
do it on legacy transtion too for solving sync issue between SysUi and
WindowManager.

Fix: 237229859
Test: manual
Test: pass existing tests
Change-Id: I8c1669a8bafa5c369ee34380aed3e2c5c0e8c326