Should fix the travis tests