diff --git a/tf-ner-poc/src/main/java/org/apache/opennlp/tf/guillaumegenthial/SequenceTagging.java b/tf-ner-poc/src/main/java/org/apache/opennlp/tf/guillaumegenthial/SequenceTagging.java
index 653b58a..637ccec 100644
--- a/tf-ner-poc/src/main/java/org/apache/opennlp/tf/guillaumegenthial/SequenceTagging.java
+++ b/tf-ner-poc/src/main/java/org/apache/opennlp/tf/guillaumegenthial/SequenceTagging.java
@@ -61,15 +61,15 @@
     FeedDictionary fd = FeedDictionary.create(tokenIds);
 
     List<Tensor<?>> run = session.runner()
-            .feed("char_ids:0", fd.getCharIdsTensor())
-            .feed("dropout:0", fd.getDropoutTensor())
-            .feed("sequence_lengths:0", fd.getSentenceLengthsTensor())
-            .feed("word_ids:0", fd.getWordIdsTensor())
-            .feed("word_lengths:0", fd.getWordLengthsTensor())
-            .fetch("proj/logits", 0)
+            .feed("chars/char_ids:0", fd.getCharIdsTensor())
+            // TODO: missing in the python code ...
+            //.feed("dropout:0", fd.getDropoutTensor())
+            .feed("words/sequence_lengths:0", fd.getSentenceLengthsTensor())
+            .feed("words/word_ids:0", fd.getWordIdsTensor())
+            .feed("chars/word_lengths:0", fd.getWordLengthsTensor())
+            .fetch("logits", 0)
             .fetch("trans_params", 0).run();
 
-
     float[][][] logits = new float[fd.getNumberOfSentences()][fd.getMaxSentenceLength()][indexTagger.getNumberOfTags()];
     run.get(0).copyTo(logits);