Update YamlDriver.php