DO NOT MERGE

Use focused task id for getting task size

When getting the selected task size, look at the focused task id, not
the running task id.
I believe this was a typo from the original change.

Bug: 190399315
Test: Local build and flash and run
Change-Id: Ice92e0562765f4f0142e27da0c38140df4c11425