Fix the broken preview in 3-button navigation mode when turning the screen off and then on. am: a322886bd6 am: b80d04eff8

Original change: https://googleplex-android-review.googlesource.com/c/platform/packages/apps/WallpaperPicker2/+/11931032

Change-Id: I5cec966bcb52f68033844babd5f7d13d3fe5cd35