В сентябре 1944 года В. И. Косолапов был произведён в капитаны и назначен на должность командира 1-й эскадрильи 594-го штурмового авиационного полка. Под его командованием в ходе Рижской и Мемельской операций его эскадрилья неоднократно оказывала реальную помощь наземным войскам при прорыве сильно укреплённых и глубоко эшелонированных рубежей противника.