From 29ef4a05fee595a0083f0080334106f30f33fa59 Mon Sep 17 00:00:00 2001
From: Niloth P <20315308+Niloth-p@users.noreply.github.com>
Date: Tue, 19 Nov 2024 15:28:30 +0530
Subject: [PATCH] tools/screenshots: Fix custom headers crash when using custom
 options.

---
 tools/screenshots/generate-integration-docs-screenshot | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tools/screenshots/generate-integration-docs-screenshot b/tools/screenshots/generate-integration-docs-screenshot
index c8b1629397..ada5cdb635 100755
--- a/tools/screenshots/generate-integration-docs-screenshot
+++ b/tools/screenshots/generate-integration-docs-screenshot
@@ -172,7 +172,8 @@ def send_bot_payload_message(
     data, json_fixture, fixture_name = get_fixture_info(fixture_path)
 
     headers = get_requests_headers(integration.name, fixture_name)
-    headers.update(config.custom_headers)
+    if config.custom_headers:
+        headers.update(config.custom_headers)
     if config.use_basic_auth:
         credentials = base64.b64encode(f"{bot.email}:{bot.api_key}".encode()).decode()
         auth = f"basic {credentials}"