That sounds… extreme. Wikipedia lists several examples [1] where a diplomat's immunity was waived by their home country since they committed offense unrelated to their function. And the same seems to be true for head states committing war crimes, as it is unrelated to their function [2].