Fix assertion in test case.