Minor unit test change