Fixed up test