fixed typo for train file