Fixed up test code.