working on unit test corrections