W październiku 1944 roku został dowódcą 3 Armii Uderzeniowej. Armia ta blokowała wojska niemieckie znajdujące się w Kurlandii, a następnie w okresie styczeń – luty 1945 uczestniczyła pod jego dowództwem w operacji wiślańsko-odrzańskiej, biorąc udział w likwidacji okrążonych niemieckich zgrupowań w Poznaniu, Schneidemühl (obecnie Piła), Deutsch Krone (obecnie Wałcz) i Arnswalde (obecnie Choszczno) oraz rozszerzeniu przyczółka na lewym brzegu Odry. W marcu 1945 roku został dowódcą 67 Armii, którą dowodził do końca wojny i uczestniczył w blokadzie Grupy Armii Kurlandia.