small unit test updates