diff --git a/src/datasets/io/sql.py b/src/datasets/io/sql.py
index d11a4ca70c0d..654d3589a1f8 100644
--- a/src/datasets/io/sql.py
+++ b/src/datasets/io/sql.py
@@ -114,7 +114,7 @@ def _write(self, conn: Connection, **to_sql_kwargs) -> int:
                 df = self._batch_sql(offset)
                 written += df.to_sql(
                     self.table_name, conn, **to_sql_kwargs, if_exists="replace" if offset == 0 else "append"
-                )
+                ) or len(df)
 
         else:
             num_rows, batch_size = len(self.dataset), self.batch_size