Remove diagram junk