Clean up test runner code