Clean up integration code