Minor unit test update