Kozlov lãnh đạo hệ thống phòng thủ của Kharkov và là một trong những người cuối cùng rời khỏi nó trước khi quân Đức tái chiếm nó vào ngày 14 tháng 3 năm 1943. Từ ngày 14 đến ngày 21 tháng 3 năm 1943, các sư đoàn Liên Xô liên tục rút lui qua khu rừng phía đông bắc Mokhnachev, và Kozlov cũng đi theo con đường đó rời khỏi Kharkov. Từ tháng 5 đến tháng 8 năm 1943, ông được giao nhiệm vụ tại Phương diện quân Leningrad và từ tháng 8 năm 1943 được bổ nhiệm làm Phó Tư lệnh Phương diện quân Zabaikal, nơi ông tham gia các cuộc tấn công của Liên Xô chống lại Nhật Bản. Từ năm 1946 cho đến khi nghỉ hưu năm 1954, ông được bổ nhiệm làm Phó Tư lệnh Quân khu Zabaikal. Ông mất năm 1967 tại Minsk.