fix diagram