unit tests changes