Fixed test for unit