С декабря 1944 года командовал 31-й армией в составе 3-го Белорусского фронта. В Восточно-Прусской наступательной операции в январе-марте 1945 года армия прорвала Хейльсбергский укрепрайон, овладела городами Ширвиндт, Лабиау, Белау, Тапиау и вышла к заливу Фришес-Хафф, отрезав пути отхода крупной немецкой группировки южнее Кёнигсберга. В марте 1945 года в ходе Хейльсбергской фронтовой операции армия штурмом взяла последний крупный укреплённый пункт немецкой обороны в этом районе Хайлигенба́йль и при ликвидации окружённой хейльсбергской группировки захватила тысячи пленных и большое количество вооружения и военной техники противника.