Fixing load_from_dir to only load yml files