Fix test failure
Update test configuration