Fixing miscaptured diagram