С июня 1944 года — командир 188-й бомбардировочной авиационной дивизии, завершил её формирование в Московском военном округе. В августе 1944 года прибыл с дивизией в 15-ю воздушную армию 2-го Прибалтийского фронта. В апреле 1945 года дивизию передали в 16-ю воздушную армию 1-го Белорусского фронта. Дивизия под его командованием принимала участие в Режицко-Двинской, Прибалтийской наступательных операциях, в блокаде Курляндской группировки противника, а завершила боевой путь в Берлинской наступательной операции. За годы войны А. И. Пушкин совершил 70 боевых вылетов.