Clean up test classes