Clean up test a little bit.