Update test assertion