Updating code due to TaskStackChangeListener changes

Bug: 221961069
Test: Presubmit
Change-Id: I40b2e5a12c4cd6faa53ee788939c31b25bf5c37b
2 files changed