TST: Fix failing tests.