Fixed up test expectations