[misc] Fixed failing tests caused by the previous commit.