fix up test