TST: fix tests