Fix task snapshots not being taken

- Regression from ag/9684093, we should be using the passed in tasks
  to take snapshots

Test: Swipe up from an app and ensure task snapshot shows
Bug: 146497751
Change-Id: I10c46d1d5063a93d4055cf252c6f254264eeda96