Clean up graph integration