Fix failing test
Fix broken test.