Clean up test-code