Refactor of broken test on travis