unbreak bc test (#51702)
Summary:
Caused by https://github.com/pytorch/pytorch/issues/48223 revert
Pull Request resolved: https://github.com/pytorch/pytorch/pull/51702
Reviewed By: mruberry
Differential Revision: D26245905
Pulled By: ngimel
fbshipit-source-id: 9fd7860ecb5c22b2e568db3347d51e648d6c5d6b
