Fixed tests breakage from last commit.