diff --git a/api/commands.py b/api/commands.py
index 8ca19e1dac..084fd576a1 100644
--- a/api/commands.py
+++ b/api/commands.py
@@ -321,6 +321,8 @@ def migrate_knowledge_vector_database():
             )
 
             datasets = db.paginate(select=stmt, page=page, per_page=50, max_per_page=50, error_out=False)
+            if not datasets.items:
+                break
         except SQLAlchemyError:
             raise