From 7947248f4619dab576c9eeabe4fda1610f08aac3 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Tue, 26 Nov 2024 14:09:26 +0530
Subject: [PATCH] test: fix test

---
 tests/documentation_tests/test_env_keys.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tests/documentation_tests/test_env_keys.py b/tests/documentation_tests/test_env_keys.py
index 9caae7fd8d2c..6b7c15e2b9de 100644
--- a/tests/documentation_tests/test_env_keys.py
+++ b/tests/documentation_tests/test_env_keys.py
@@ -57,7 +57,9 @@
 
         # Find the section titled "general_settings - Reference"
         general_settings_section = re.search(
-            r"### environment variables - Reference(.*?)###", content, re.DOTALL
+            r"### environment variables - Reference(.*?)(?=\n###|\Z)",
+            content,
+            re.DOTALL | re.MULTILINE,
         )
         print(f"general_settings_section: {general_settings_section}")
         if general_settings_section: