fix diagram of the first step