Fix output dir bug