Fix module name to pyyaml