It would seem like a good idea to set /var/run/reboot-required if a reboot is required, no? This would mirror the logic apt-get establishes.