TST fix tests