[UnitTests] Fixed failing unit test.