De octubre de 1944 a marzo de 1945 sirvió como comandante del 3.º Ejército de Choque del 2º Frente Báltico (en diciembre de 1944, el ejército fue transferido al 1er Frente Bielorruso). con dicho ejército participó en la Ofensiva de Riga (14 de septiembre - 22 de octubre de 1944), que permitió expulsar a la Wehrmacht de Letonia y liberar Riga. Posteriormente participó en el bloqueo del Grupo de Ejército Curlandia, formado por las restos del antiguo Grupo de Ejército Norte cercados en la Bolsa de Curlandia, y finalmente, tomo parte en la ofensiva del Vístula-Óder (enero-febrero de 1945) y en la ofensiva de Pomerania Oriental (febrero-abril de 1945).