TST: Fix failing tests

See #465