Fixed test a lil.