clean up test a bit