diff --git a/surfsense_backend/app/agents/new_chat/tools/notion/update_page.py b/surfsense_backend/app/agents/new_chat/tools/notion/update_page.py
index 088041add..890f3fefe 100644
--- a/surfsense_backend/app/agents/new_chat/tools/notion/update_page.py
+++ b/surfsense_backend/app/agents/new_chat/tools/notion/update_page.py
@@ -220,7 +220,7 @@ def create_update_notion_page_tool(
                 f"update_page result: {result.get('status')} - {result.get('message', '')}"
             )
 
-            if result.get("status") == "success":
+            if result.get("status") == "success" and document_id is not None:
                 from app.services.notion import NotionKBSyncService
 
                 logger.info(f"Updating knowledge base for document {document_id}...")
diff --git a/surfsense_backend/app/services/notion/kb_sync_service.py b/surfsense_backend/app/services/notion/kb_sync_service.py
index c51de09b9..3cacdcfe5 100644
--- a/surfsense_backend/app/services/notion/kb_sync_service.py
+++ b/surfsense_backend/app/services/notion/kb_sync_service.py
@@ -83,7 +83,7 @@ class NotionKBSyncService:
                     f"Appended IDs (first 3): {appended_block_ids[:3]}, Fetched IDs count: {len(fetched_block_ids)}"
                 )
 
-                if len(found_blocks) >= len(appended_block_ids) * 0.8: # 80% threshold
+                if len(found_blocks) >= len(appended_block_ids) * 0.8:  # 80% threshold
                     logger.info(
                         f"Content verified fresh: found {len(found_blocks)}/{len(appended_block_ids)} appended blocks"
                     )