17 czerwca 1944 dywizja ta została przemianowana na 99 Gwardyjską Dywizję Piechoty, a w grudniu 1944 na 99 Dywizję Powietrznodesantową. 2 listopada 1944 otrzymał stopień generała majora. W lutym 1945 dowodzona przez niego dywizja została włączona w skład 9 Gwardyjskiej Armii 3 Frontu Ukraińskiego, biorąc udział w zajęciu Węgier, Austrii i Czechosłowacji w ramach operacji budapesztańskiej, wiedeńskiej i bratysławsko-brneńskiej. Za umiejętne dowodzenie dywizją podczas działań bojowych na terytorium Węgier i Austrii w marcu 1945 został przedstawiony do odznaczenia tytułem Bohatera Związku Radzieckiego. 23 kwietnia 1945 podczas działań wojennych w Austrii został ciężko ranny po wybuchu miny i w nocy na 24 kwietnia zmarł. Został pochowany na Cmentarzu Nowodziewiczym. 28 kwietnia 1945 pośmiertnie otrzymał tytuł Bohatera Związku Radzieckiego i Order Lenina. Jego imieniem nazwano ulicę w Mariupolu.