TST: fix unit tests