Fix small regression in last commit.