Correcting unit test