try to fix tests :'(