Fixes test assertion.