diff --git a/Train.py b/Train.py
index a15c6007ccd967d8f776aabe46d0cf799bf199aa..ca196577125a532909212daf27bfb76c53e1407d 100644
--- a/Train.py
+++ b/Train.py
@@ -210,7 +210,7 @@ def trainModelRl(debug, networkName, modelDir, filename, nbIter, batchSize, devF
   if os.path.isfile(modelDir+"/dicts.json") :
     dicts.load(modelDir+"/dicts.json")
   else :
-    dicts.readConllu(filename, ["FORM","UPOS","LETTER","LEXICON"], 2, pretrained)
+    dicts.readConllu(filename, Networks.getNeededDicts(networkName), 2, pretrained)
     dicts.addDict("HISTORY", transitionNames)
     dicts.save(modelDir + "/dicts.json")