(defun ~/org/update-cookies ()
  (org-update-statistics-cookies 'all))

(defun ~/org/setup ()
  (add-hook 'before-save-hook #'~/org/update-cookies nil t))

(add-hook 'org-mode-hook #'~/org/setup)