Fixed up test assertion