Fixed issue in yaml loader