Do you have timeoutstopseconds set? I guess I should ask if you're using a systemd unit first 😁

It should wait for it on a reboot if you have the settings set and it's running in the system.slice