Fix test breakage introduced in last commit.