Fix issues with files missing