Fix regression in last commit.