TST: Fix failing test.
BUG: fix another failing test.