Fixed test that was returning null pointer exception