During March, the Japanese launched a major offensive (codenamed Operation U-Go) into Manipur. Having been withdrawn from the Arakan battles, the division proceeded by road and rail to Dimapur, where it came under command of XXXIII Corps and took part in the Battle of Kohima. The 161st Indian Brigade, part of the 5th Indian Division, came under command, while the 89th Brigade was flown to Imphal, to replace 161st Brigade in 5th Division. During early May, 33rd Brigade completed the recapture of Kohima Ridge, while the main body of the division recaptured Naga Village to the north of the ridge. During the later part of the month and early June, the division advanced through heavy monsoon rains along rough tracks to the east of the main road from Kohima to Imphal, and cleared Japanese stragglers from Ukhrul.