free_extent_buffer(root->commit_root);
                free_extent_buffer(leaf);
        }
-       kfree(root);
+       btrfs_put_fs_root(root);
 
        return ERR_PTR(ret);
 }
        leaf = btrfs_alloc_tree_block(trans, root, 0, BTRFS_TREE_LOG_OBJECTID,
                        NULL, 0, 0, 0);
        if (IS_ERR(leaf)) {
-               kfree(root);
+               btrfs_put_fs_root(root);
                return ERR_CAST(leaf);
        }
 
        return root;
 
 find_fail:
-       kfree(root);
+       btrfs_put_fs_root(root);
 alloc_fail:
        root = ERR_PTR(ret);
        goto out;
 {
        kfree(fs_info->balance_ctl);
        kfree(fs_info->delayed_root);
-       kfree(fs_info->extent_root);
-       kfree(fs_info->tree_root);
-       kfree(fs_info->chunk_root);
-       kfree(fs_info->dev_root);
-       kfree(fs_info->csum_root);
-       kfree(fs_info->quota_root);
-       kfree(fs_info->uuid_root);
-       kfree(fs_info->free_space_root);
+       btrfs_put_fs_root(fs_info->extent_root);
+       btrfs_put_fs_root(fs_info->tree_root);
+       btrfs_put_fs_root(fs_info->chunk_root);
+       btrfs_put_fs_root(fs_info->dev_root);
+       btrfs_put_fs_root(fs_info->csum_root);
+       btrfs_put_fs_root(fs_info->quota_root);
+       btrfs_put_fs_root(fs_info->uuid_root);
+       btrfs_put_fs_root(fs_info->free_space_root);
+       btrfs_put_fs_root(fs_info->fs_root);
        kfree(fs_info->super_copy);
        kfree(fs_info->super_for_commit);
-       btrfs_put_fs_root(fs_info->fs_root);
        kvfree(fs_info);
 }
 
        if (IS_ERR(log_tree_root->node)) {
                btrfs_warn(fs_info, "failed to read log tree");
                ret = PTR_ERR(log_tree_root->node);
-               kfree(log_tree_root);
+               btrfs_put_fs_root(log_tree_root);
                return ret;
        } else if (!extent_buffer_uptodate(log_tree_root->node)) {
                btrfs_err(fs_info, "failed to read log tree");
                free_extent_buffer(log_tree_root->node);
-               kfree(log_tree_root);
+               btrfs_put_fs_root(log_tree_root);
                return -EIO;
        }
        /* returns with log_tree_root freed on success */
                btrfs_handle_fs_error(fs_info, ret,
                                      "Failed to recover log tree");
                free_extent_buffer(log_tree_root->node);
-               kfree(log_tree_root);
+               btrfs_put_fs_root(log_tree_root);
                return ret;
        }
 
 
        clear_extent_bits(&log->dirty_log_pages, 0, (u64)-1,
                          EXTENT_DIRTY | EXTENT_NEW | EXTENT_NEED_WAIT);
        free_extent_buffer(log->node);
-       kfree(log);
+       btrfs_put_fs_root(log);
 }
 
 /*
                                                        log->node->len);
                        free_extent_buffer(log->node);
                        free_extent_buffer(log->commit_root);
-                       kfree(log);
+                       btrfs_put_fs_root(log);
 
                        if (!ret)
                                goto next;
                btrfs_put_fs_root(wc.replay_dest);
                free_extent_buffer(log->node);
                free_extent_buffer(log->commit_root);
-               kfree(log);
+               btrfs_put_fs_root(log);
 
                if (ret)
                        goto error;
        free_extent_buffer(log_root_tree->node);
        log_root_tree->log_root = NULL;
        clear_bit(BTRFS_FS_LOG_RECOVERING, &fs_info->flags);
-       kfree(log_root_tree);
+       btrfs_put_fs_root(log_root_tree);
 
        return 0;
 error: