Fix broken travis test