diff --git a/tf-ner-poc/src/main/python/namefinder.py b/tf-ner-poc/src/main/python/namefinder.py
index 3bf8405..cd5a464 100644
--- a/tf-ner-poc/src/main/python/namefinder.py
+++ b/tf-ner-poc/src/main/python/namefinder.py
@@ -343,7 +343,7 @@
 
     char_dict = {k: v for v, k in enumerate(char_set | char_set_dev)}
 
-    embedding_ph, token_ids_ph, char_ids_ph, word_lengths_ph, sequence_lengths_ph, labels_ph, train_op \
+    embedding_ph, token_ids_ph, char_ids_ph, word_lengths_ph, sequence_lengths_ph, labels_ph, dropout_keep_prob, train_op \
         = name_finder.create_graph(len(char_set | char_set_dev), embeddings)
 
     write_mapping(word_dict, 'word_dict.txt')