Fix output dir selection