clean up test code