Fix test fails