Fix broken test for #124