u64 end = start + PAGE_SIZE - 1;
        size_t blocksize = page->mapping->host->i_sb->s_blocksize;
 
+       /* This function is only called for the btree inode */
+       ASSERT(tree->owner == IO_TREE_BTREE_INODE_IO);
+
        start += ALIGN(offset, blocksize);
        if (start > end)
                return 0;
 
        lock_extent_bits(tree, start, end, &cached_state);
        wait_on_page_writeback(page);
-       clear_extent_bit(tree, start, end, EXTENT_LOCKED | EXTENT_DELALLOC |
-                        EXTENT_DO_ACCOUNTING, 1, 1, &cached_state);
+
+       /*
+        * Currently for btree io tree, only EXTENT_LOCKED is utilized,
+        * so here we only need to unlock the extent range to free any
+        * existing extent state.
+        */
+       unlock_extent_cached(tree, start, end, &cached_state);
        return 0;
 }