Simplify test assertion