[shame] fix the tests