fix bc breakage of #52043 (#55303)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/55303
Test Plan: Imported from OSS
Reviewed By: walterddr
Differential Revision: D27567671
Pulled By: agolynski
fbshipit-source-id: 771e75b68be52dd5dd31437238d1f9fef481f853
