Однако Гитлер ничего не забыл. Став жертвой газовой атаки в Первой мировой войне, Хейнрици с тех пор страдал различными болезнями желудка. Через несколько месяцев после инцидента с Герингом Гитлер, сославшись на эти болезни, внес Хейнрици в список командного состава, не состоящего на действительной службе, по причине «слабого здоровья». Его отправили в отставку и, по словам самого Хейнрици, «просто забыли» в санатории для послебольничного долечивания в Карлсбаде в Чехословакии. Через несколько недель после его увольнения из армии русские в первый раз прорвались через позиции его бывшего соединения — 4-й армии.
В первые месяцы 1944 года Хейнрици оставался в Карлсбаде и издалека наблюдал за трагическими событиями, медленно превращавшими империю Гитлера в руины: июньским вторжением западных союзников в Нормандию, продвижением англо-американских войск вверх по «итальянскому сапогу» и захватом Рима, неудачным покушением на Гитлера 20 июля, неодолимым шествием русских по Восточной Европе. По мере того как ситуация становилась все более критической, Хейнрици все тяжелее переносил свое вынужденное бездействие. Он мог бы обратиться к Гитлеру с просьбой о назначении, но не хотел унижаться.
Наконец поздним летом сорок четвертого года после восьми месяцев вынужденной отставки Хейнрици вновь призвали на службу, на этот раз в Венгрию, и поручили командование оказавшимися в тяжелом положении 1-й танковой и 1-й венгерской армиями.
В Венгрии Хейнрици взялся за старое. В разгар сражений генерал-полковник Фердинанд Шернер, протеже Гитлера и начальник Хейнрици в Венгрии, издал следующую директиву: «…любой солдат, обнаруженный без приказа вдали от передовой, должен быть немедленно казнен, а тело его выставлено в назидание остальным». Хейнрици, возмущенный этим приказом, сердито возразил: «Подобные методы никогда не применялись под моим командованием и никогда не будут применяться».
Хейнрици был вынужден отступить из Северной Венгрии в Чехословакию. Он сражался за каждый клочок земли так отчаянно, что его наградили мечами к дубовым листьям его Рыцарского креста — удивительное достижение для человека, которого так сильно не любил сам Гитлер. О награждении генералу сообщили 3 марта 1945 года, а сейчас, всего две недели спустя, он мчался в Цоссен с приказом в кармане принять командование группой армий «Висла».
Глядя, как мощный «мерседес» пожирает колесами ленту имперского шоссе 96, Хейнрици задавался вопросом, куда же это шоссе в конце концов приведет его. Он вспомнил реакцию своих штабистов в Венгрии на известие о новом назначении и вызове к генералу Хейнцу Гудериану, главнокомандующему сухопутными войсками в штаб ОКХ (Oberkommando der Heers). Они были в шоке. «Вам действительно нужна эта работа?» — спросил его начальник штаба.
