Fixing missing dir issue