Fixes flaky tests

The CountDownLatch should be put at the last line to avoid making
CountDownLatch#await passed without updating values.

Bug: 224717753
Test: atest WindowMagnificationControllerTest
Change-Id: I29530a02780291d684a30e31b4811f2b95fe0faf
2 files changed