Clean up legend comment