Fix PyYAML dependency for pypi