Fix test accordingly.