Clean up test controller