diff --git a/tools/context.py b/tools/context.py
index 579fe7f4..31a2b341 100644
--- a/tools/context.py
+++ b/tools/context.py
@@ -10,7 +10,7 @@
 import os
 import stat
 from pickle import dumps
-import pyaml
+import yaml
 from jinja2 import pass_context
 
 from . import macros
@@ -390,7 +390,7 @@ def write_manifest(context, manifest='manifest.yaml', allocator="allocators.pick
 
             data = {"cap_symbols": stash.cap_symbols, "region_symbols": stash.region_symbols}
 
-            manifest_file.write(pyaml.dump(data))
+            manifest_file.write(yaml.dump(data))
             allocator_file.write(dumps(stash.allocator_state))
         return ""