diff --git a/fs/btrfs/inode.c b/fs/btrfs/inode.c
index 8f0376a4efb7..491c2b7f48bf 100644
--- a/fs/btrfs/inode.c
+++ b/fs/btrfs/inode.c
@@ -5305,11 +5305,7 @@ static void evict_inode_truncate_pages(struct inode *inode)
 		clear_bit(EXTENT_FLAG_LOGGING, &em->flags);
 		remove_extent_mapping(map_tree, em);
 		free_extent_map(em);
-		if (need_resched()) {
-			write_unlock(&map_tree->lock);
-			cond_resched();
-			write_lock(&map_tree->lock);
-		}
+		cond_resched_rwlock_write(&map_tree->lock);
 	}
 	write_unlock(&map_tree->lock);