UnitTest fixed