Further diagram correction