Minor fix in the test assertion.