Minor UnitTest fix