W marcu 1944 został dowódcą 4 Armii Pancernej (17 marca 1945 została przemianowana na 4 Gwardyjską Armię Pancerną), którą dowodził do końca wojny. Dowodząc to armią wziął udział w oswobodzeniu prawobrzeżnej Ukrainy, operacji lwowsko-sandomierskiej, wschodniokarpackiej, wiślańsko-odrzańskiej, śląskiej i berlińskiej.