Summary: This needs to be on the new TransferData() as well as the old one.

Test Plan: Manual.

Reviewers: #stirling, yzhao

Reviewed By: #stirling, yzhao

Differential Revision: https://phab.corp.pixielabs.ai/D8428

GitOrigin-RevId: 12bf5dc8b2d858f17b969a7d3355538b1fbec3cd