Fix pyyaml