A bit less weird
Fix failing test