Fixed YamlImporter working with datasets