Adjust to changes in kmap_atomic