unit test updates