Fix tests, I guess.