Fixed bug in yield_files