Add support for worker restarting