Clean up test codes