In January 1945, the 332nd finally left 4th Shock, and was reassigned to 6th Guards Army, still in 1st Baltic Front. In 1945 the Soviet commands in the southern Baltic States were in constant flux, as both the assault on Königsberg and the reduction or containment of the Courland Pocket were attempted simultaneously. In March the division was moved to 42nd Army, facing the Courland group. In the last month of the war the division was in 23rd Guards Rifle Corps, in 67th Army of Leningrad Front on the Gulf of Riga.