diff --git a/distributed/main_sharding_config.py b/distributed/main_sharding_config.py
index 6254d01efc35f2eb5635152a9a280d86fc8902ef..bd9661eec1124da6aa33297d28f869c42a0c19a3 100644
--- a/distributed/main_sharding_config.py
+++ b/distributed/main_sharding_config.py
@@ -172,13 +172,13 @@ class Format:
     def load(cls):
         with open(PathConf.get_src_yaml(), "rt", encoding='utf-8') as f:
             count = f.read()
-            return yaml.load(count)
+            return yaml.safe_load(count)
 
     @classmethod
     def load_user_input(cls):
         with open(PathConf.get_user_input_yaml(), "rt", encoding='utf-8') as f:
             count = f.read()
-            return yaml.load(count)
+            return yaml.safe_load(count)
 
     @classmethod
     def modified_output(cls):