[UnitTests] Fixed unit test.