TST: Fix broken tests that got merged in #139

Sorry team!