W lipcu 1944 roku został dowódcą 33 Armii, na czele której wziął udział w forsowaniu rzeki Niemen, zdobyciu Kowna. W październiku 1944 roku odwołany ze stanowiska dowódcy armii, został dowódcą 6 Gwardyjskiego Korpusu Strzeleckiego i uczestniczył w forsowaniu Dunaju w rejonie Mohacza, następnie uczestniczył w operacji budapeszteńskiej.