UnitTest bugfix