Fix incorrect test assertion