diff --git a/fs/iomap/buffered-io.c b/fs/iomap/buffered-io.c
index 25448d5827d259d63743e8cefc042b2d5338fe4e..73524df884bb4c87ea023d10fb74592d34b1ff2e 100644
--- a/fs/iomap/buffered-io.c
+++ b/fs/iomap/buffered-io.c
@@ -1073,7 +1073,7 @@ iomap_page_mkwrite_actor(struct inode *inode, loff_t pos, loff_t length,
 		block_commit_write(page, 0, length);
 	} else {
 		WARN_ON_ONCE(!PageUptodate(page));
-		set_page_dirty(page);
+		iomap_set_range_dirty(page, offset_in_page(pos), length);
 	}
 
 	return length;