Small tweak to diagram.