Fix breakage from last commit.