diff --git a/models/hyperparameters/params_dict.py b/models/hyperparameters/params_dict.py
index 1eb247957..beaeeddac 100644
--- a/models/hyperparameters/params_dict.py
+++ b/models/hyperparameters/params_dict.py
@@ -292,7 +292,7 @@ def _get_kvs(tokens, params_dict):
 def read_yaml_to_params_dict(file_path):
   """Reads a YAML file to a ParamsDict."""
   with tf.gfile.Open(file_path, 'r') as f:
-    params_dict = yaml.load(f)
+    params_dict = yaml.safe_load(f)
     return ParamsDict(params_dict)