Restart agent after an upgrade