Fix to the travis test