Summary: Handle naming change

Test Plan: Run lint

Reviewers: zasgar, michelle

Reviewed By: michelle

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

GitOrigin-RevId: 0e44dd17b6a00aa9a8e2dee1880ef13a1fedbf84