Fix broken test!