Fix minor compile issue after MFC r347288.
This is a direct commit.
Sponsored by: Mellanox Technologies