fix bug train file input