Fixed bug in yaml