fix one of tests