В июне 1944 года по предложению прежнего командующего армией генерал-полковника танковых войск П. С. Рыбалко вновь убыл на фронт на должность заместителя начальника штаба 3-й гвардейской танковой армии 1-го Украинского фронта. Участвовал с ней в Львовско-Сандомирской наступательной операции. На заключительном этапе операции в боях за город Львов непосредственно командовал ударной группой армии, атаковавшей противника с тыла. В августе армия вела бои на сандомирском плацдарме, затем 7 сентября выведена в резерв Ставки ВГК. 28 октября она вновь вошла в состав 1-го Украинского фронта и участвовала в Сандомирско-Силезской, Нижне-Силезской, Берлинской и Пражской наступательных операциях.