Fix test failed