diff --git a/utils.py b/utils.py
index a0e4349..6e269d6 100644
--- a/utils.py
+++ b/utils.py
@@ -12,7 +12,7 @@
 
 def get_config(config):
     with open(config, 'r') as stream:
-        return yaml.load(stream)
+        return yaml.safe_load(stream)
 
 def prepare_sub_folder(output_directory):
     image_directory = os.path.join(output_directory, 'images')