diff --git a/scripts/docs_examples_generator/config.py b/scripts/docs_examples_generator/config.py
index 814e6f92..1a7f931a 100644
--- a/scripts/docs_examples_generator/config.py
+++ b/scripts/docs_examples_generator/config.py
@@ -7,7 +7,7 @@
 dirname = Path(__file__).parent.absolute()
 dotenv.load_dotenv()
 
-_branch_name = os.environ["BRANCH_NAME"]
+_branch_name = os.environ.get("BRANCH_NAME", "main")
 
 
 class GeneratorConfig(BaseModel):