Fix missing file issues