С января 1944 года до конца войны — командующий 10-й гвардейской армией на 2-м Прибалтийском и Ленинградском фронтах. Во главе этой армии проявил себя успешным командующим, действуя в сложных природных условиях Прибалтики (леса, болота, обилие водных преград). Армия добилась значительных успехов в Режицко-Двинской, Мадонской, Рижской наступательных операциях 1944 года. С октября 1944 до мая 1945 года части армии вместе с другими войсками фронта блокировали Курляндскую группировку и предпринимали несколько попыток уничтожить её, которые из-за недостатка сил не увенчались успехом. Войска противника удалось потеснить только на несколько десятков километров, но они продолжали обороняться до общей капитуляции Германии и сдались в плен только 9 мая 1945 года.