diff --git a/src/migrator/full_migration.py b/src/migrator/full_migration.py
index 200b888..98966e6 100644
--- a/src/migrator/full_migration.py
+++ b/src/migrator/full_migration.py
@@ -198,9 +198,10 @@ class FullMigrator:
                                         batch_count += 1
                                         progress.update(fetched_in_buffer)
                                         # Update migration state after every batch flush
-                                        # Do NOT set last_completed_partition yet - partition is still being processed
+                                        # Keep last_completed_partition if we've completed partitions before
                                         self._update_migration_state(
-                                            pg_conn, migrated, None, migration_start_time
+                                            pg_conn, migrated, None, migration_start_time,
+                                            last_partition=last_processed_partition
                                         )
                                         logger.debug(
                                             f"Partition {partition}: flushed {inserted} rows, "
@@ -217,9 +218,10 @@ class FullMigrator:
                                     migrated += inserted
                                     batch_count += 1
                                     progress.update(fetched_in_buffer)
-                                    # Still don't set last_completed_partition - partition is still being finalized
+                                    # Keep last_completed_partition if we've completed partitions before
                                     self._update_migration_state(
-                                        pg_conn, migrated, None, migration_start_time
+                                        pg_conn, migrated, None, migration_start_time,
+                                        last_partition=last_processed_partition
                                     )
                                     logger.debug(
                                         f"Partition {partition} final flush: {inserted} rows, "