Repair test breakage from previous commit.