Fin 1944, Lehmann est transféré au Corps Panzer SS de la SS Panzer Armee en tant que chef d'état-major. Il restera à ce poste jusqu'en , quand il reçoit le commandement de la SS Das Reich et dirige la Division à travers les batailles autour de Vienne pendant l'Opération Frühlingserwachen.