Better test assertion