Po zakończeniu leczenia w dniu 16 stycznia 1945 roku został dowódcą 26 Armii, która w ramach 3 Frontu Ukraińskiego wzięła udział w oblężeniu Budapesztu, operacji balatońskiej, wiedeńskiej i w dalszych walkach na terenie Austrii.