Fix compile issue with the last commit.