HP Image Assistant - Install Updates Task - Handling for code 3010 (machine needs to reboot)

Currently, if you use the HP Image Assistant - Install Updates Task with the “do not reboot flag”, the install tries to run 3 times. Each time it “fails” with 3010 and retries. That code translate to “the machine needs to reboot”. so it seems like the install actually worked! Can something be added to the script to handle success and pending reboot?
Also, it does not appear that Image Assistant sets the machine to pending reboot, so reboots are not happening. Can Immy’s set-pendingreboot be tied in?