UnitTest fix