Any chance you could create a PR with this?

Seems not easy to do at the moment without messing up my existing repository. This will be together in my upcoming PR.

But in the moment if anyone wants to make the change, the included code is the only modification needed, unless you would also want to remove the linuxSafeRestart parameters, there would be a couple other small changes.