diff --git a/tools/codegen/core/gen_config_vars.py b/tools/codegen/core/gen_config_vars.py
index 43b48871a7b..1f5f02ed1f2 100755
--- a/tools/codegen/core/gen_config_vars.py
+++ b/tools/codegen/core/gen_config_vars.py
@@ -29,7 +29,7 @@ import sys
 import yaml
 
 with open('src/core/lib/config/config_vars.yaml') as f:
-    attrs = yaml.load(f.read(), Loader=yaml.FullLoader)
+    attrs = yaml.safe_load(f.read(), Loader=yaml.FullLoader)
 
 error = False
 today = datetime.date.today()
diff --git a/tools/codegen/core/gen_stats_data.py b/tools/codegen/core/gen_stats_data.py
index f1e979760d0..f4710ecf230 100755
--- a/tools/codegen/core/gen_stats_data.py
+++ b/tools/codegen/core/gen_stats_data.py
@@ -25,7 +25,7 @@ import sys
 import yaml
 
 with open('src/core/lib/debug/stats_data.yaml') as f:
-    attrs = yaml.load(f.read(), Loader=yaml.Loader)
+    attrs = yaml.safe_load(f.read(), Loader=yaml.Loader)
 
 REQUIRED_FIELDS = ['name', 'doc']