После излечения был направлен на учёбу и в мае 1944 года окончил Курсы усовершенствования высшего начальствующего состава при Высшей военной академии имени К. Е. Ворошилова. В августе 1944 года был направлен в распоряжение Военного совета 3-го Белорусского фронта. 30 августа он был назначен заместителем командира 16-го гвардейского стрелкового корпуса. А 19 октября 1944 года, когда бои перенеслись на уже территорию Восточной Пруссии (в разгар Гумбиннен-Гольдапской наступательной операции), вступил в командование 16-й гвардейской стрелковой дивизией 36-го гвардейского стрелкового корпуса 11-й гвардейской армии этого фронта. Командовал этой дивизией до конца войны.