diff --git a/ci_scripts/postprocess_toc_yml.py b/ci_scripts/postprocess_toc_yml.py
index cda06d31f5fb..580c2d77f579 100644
--- a/ci_scripts/postprocess_toc_yml.py
+++ b/ci_scripts/postprocess_toc_yml.py
@@ -21,7 +21,7 @@ def rewrite_yml(data):
 
 with open("toc.yml", 'r') as stream:
     try:
-        data_loaded = yaml.load(stream)
+        data_loaded = yaml.safe_load(stream)
         for node in data_loaded:
             if 'name' in node:
                 if node['name'].startswith('azure.') and node['name'] not in skipped_level2_packages: