Clean up test code style.