Clean up integration tests