improv(hooks): improving the update hook