Fix bug with output folder