Od 13. marca 1944 velil 33. armáde Západného frontu, už 12. apríla bol poverený vedením 2. bieloruského frontu a od 6. augusta 1944 bol veliteľom 4. ukrajinského frontu. 6. októbra 1944 mu bola navrátená hodnosť armádny generál. V marci 1945 sa stal náčelníkom štábu 1. ukrajinského frontu, v rámci ktorého sa podieľal na oslobodzovaní Česko-Slovenska (Pražská operácia).