Fix yaml parsing bug