Try fix falling unit test on travis