unit test corrections