Fix the directory issue