Fix broken test ##