cleaning up test code