Correct unit test assertion.