Fix the travis tests