Fixed test breaking in Travis