Enhancement: improve test assertion