Ende Oktober 1943 übernahm er, nachdem sich die Division aus der Ukraine zurückziehen musste, das Kommando über die 340. Infanterie-Division. Diese wurde Ende 1943 zerschlagen und anschließend wieder aufgebaut. Am 1. Januar 1944 wurde Ehrig Generalmajor. Im April 1944 wurde die Division bei Brody eingeschlossen, konnte sich aber befreien. Mitte Juni 1944 gab er das Kommando an den Generalmajor Otto Beutler ab. Anfang Juli 1944 wurde er Kommandeur der neu aufgestellten Schattendivision Infanterie-Division Grafenwöhr, welche Mitte Juli 1944 zur Aufstellung der 544. Grenadier-Division verwendet wurde. Ehrig wurde Kommandeur der 544. Grenadier-Division, ebenso wie auch der aus der 544. Grenadier-Division hervorgegangenen 544. Volksgrenadier-Division. Seine Beförderung zum Generalleutnant erfolgte am 9. November 1944. Ab Anfang Januar 1945 war die 544. Volksgrenadier-Division im Verbund mit der 17. Armee in der Weichsel-Oder-Operation und der Westkarpatische Operation eingebunden. Die Division geriet im Mai 1945 in Böhmen in sowjetische Kriegsgefangenschaft.