Post-update reboot

I’m messing around with Mender for my thesis and so far, I successfully authorized and updated my device in a demo environment, using a directory Update Module. Next is to reboot the device after the update has completed. I’m aware of the ArtifactReboot state but can’t seem to figure out how to set the NeedsArtifactReboot flag to “Automatic”. Is it during the artifact creation? If so, what’s the argument to use? If not, how else can I tell my device to reboot after the update completes? Any help would be much appreciated!

You need to set this in your Update Module. You can take a look at the following as a reference,