Cleanup.
Refactor: use XCTest expectations for asynchronous test.