Update the async storage code