Update to follow TransportMediator API changes.

Change-Id: Icedd0fbe15cf84ff4d3f0c59826f698e92138b47
2 files changed