move this to the update hook.