Fix failing test
Fix failing test