Fix test. I guess.