[UnitTest] Added missing assertion.