fix tests lol