correcting unit tests.