improve test assertion