Fix that the preview should keep the same layer when switching the screen on/off.

Action:
Move the restore state action to the previous cycle.

Bug: 259213786
Test: manual test
Change-Id: Iff49d3774824e7cfe3912a4d18fd535083cf923a