После излечения в мае 1944 года назначен на должность заместителя командира 2-го гвардейского танкового корпуса, а в сентябре того же года — на должность командира 18-го танкового корпуса, который вскоре принимал участие в боевых действиях в ходе Дебреценской и Будапештской наступательных операций. За отличия при освобождении Будапешта корпусу присвоено почётное наименование «Будапештский». Вскоре корпус участвовал в Балатонской оборонительной и Венской наступательной операциях. За образцовое выполнение заданий командования при освобождении городов Секешфехервар, Мор, Веспрем корпус был награждён орденом Суворова 3 степени, а за доблесть и мужество, проявленные при освобождении Вены — орденом Кутузова 2 степени.