Fix tests broken in previous commit.