Fix yaml problem