fix conll train script