Fix test properly.