correcting unit tests