(bad diff) Wrap with page load guard