Fix broken test on travis