Находился на излечении в госпитале в Москве. В феврале 1944 года направлен на учёбу, а в сентябре окончил ускоренный курс Высшей военной академии имени К. Е. Ворошилова. В сентябре назначен командиром 18-й гвардейской стрелковой дивизии 11-й гвардейской армии 3-го Белорусского фронта. С января 1945 года — заместитель командира 16-го гвардейского стрелкового корпуса, на этом посту участвовал в Восточно-Прусской наступательной операции, в разгроме Инстербургской группировки противника, в блокировании и штурме Кёнигсберга. С 23 (по другим данным с 27) апреля 1945 года — командир 16-й гвардейской стрелковой дивизии (36-й гвардейский стрелковый корпус, 11-я гвардейская армия, 3-й Белорусский фронт).